Type Members

1.  sealed trait LoopState extends AnyRef
2.  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 with Product with Serializable
3.  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 with Product with Serializable
4.  case class SimpleThenSWALoopState(simple: SimpleLoopState, swa: Option[SWALoopState]) extends LoopState with Product with Serializable
5.  case class TensorDescriptor(dims: Seq[Long], dataType: Byte, byteOffset: Long, byteLength: Long) extends Product with Serializable
6.  case class TensorList(tensors: Seq[TensorDescriptor], location: String, byteOffset: Long, byteLength: Long) extends Product with Serializable