SWALoopState
lamp.data.schemas.SWALoopState
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
Attributes
- Graph
-
- Supertypes
-
trait Serializabletrait Producttrait Equalstrait LoopStateclass Objecttrait Matchableclass Any
Members list
In this article