SWA
object SWA
Type members
Classlikes
Value members
Concrete methods
def epochs[I, M <: GenericModule[I, Variable] : Load, LRState, BatchStreamState, BatchStreamBuffers](model: SupervisedModel[I, M], optimizerFactory: Seq[(STen, PTag)] => Optimizer, trainBatchesOverEpoch: TrainingLoopContext => BatchStream[I, BatchStreamState, BatchStreamBuffers], validationBatchesOverEpoch: Option[TrainingLoopContext => BatchStream[I, BatchStreamState, BatchStreamBuffers]], epochs: Int, trainingCallback: TrainingCallback, validationCallback: ValidationCallback, checkpointState: Option[(SWALoopState, LRState) => IO[Unit]], validationFrequency: Int, logger: Option[Logger], learningRateSchedule: SWALearningRateSchedule[LRState], prefetch: Boolean, dataParallelModels: Seq[SupervisedModel[I, M]], initState: Option[SWALoopState], accumulateGradientOverNBatches: Int, learningRateScheduleInitState: Option[LRState], forwardPassAfterTraining: Boolean): IO[(SupervisedModel[I, M], List[(Int, Double, Option[Double])])]