lamp.data.schemas
package lamp.data.schemas
Type members
Classlikes
case class SWALoopState(model: TensorList, optimizer: TensorList, epoch: Int, lastValidationLoss: Option[Double], minValidationLoss: Option[Double], numberOfAveragedModels: Int, averagedModels: Option[TensorList], learningCurve: List[(Int, Double, Option[Double])]) extends LoopState
case class SimpleLoopState(model: TensorList, optimizer: TensorList, epoch: Int, lastValidationLoss: Option[Double], minValidationLoss: Option[Double], minValidationLossModel: Option[(Int, TensorList)], learningCurve: List[(Int, Double, Option[Double], Option[Double])]) extends LoopState
case class SimpleThenSWALoopState(simple: SimpleLoopState, swa: Option[SWALoopState]) extends LoopState
case class TensorList(tensors: Seq[TensorDescriptor], location: String, byteOffset: Long, byteLength: Long)
- Companion:
- object