Type Members

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

   dataType is pytorch scalartype:

   dataType is pytorch scalartype:

   • 1 I8
   • 2 I16
   • 3 I32
   • 4 I64
   • 5 FP16
   • 6 FP32
   • 7 FP64
   • 11 bool
   • 16 BF16
7.  case class TensorList(tensors: Seq[TensorDescriptor], location: String, byteOffset: Long, byteLength: Long) extends Product with Serializable