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