SWALoopState
lamp.data.SWALoopState
case class SWALoopState(model: Seq[STen], optimizer: Seq[STen], epoch: Int, lastValidationLoss: Option[Double], minValidationLoss: Option[Double], numberOfAveragedModels: Int, averagedModels: Option[Seq[Tensor]], learningCurve: List[(Int, Double, Option[Double])]) extends LoopState
Attributes
- Graph
-
- Supertypes
-
trait Serializabletrait Producttrait Equalstrait LoopStateclass Objecttrait Matchableclass Any
Members list
In this article