lamp-data
API
Generated with
lamp-data/lamp.data/SWA

SWA

lamp.data.SWA$
object SWA

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Self type
SWA.type

Members list

Type members

Classlikes

object SWALearningRateSchedule

Attributes

Companion
trait
Supertypes
class Object
trait Matchable
class Any
Self type
trait SWALearningRateSchedule[State]

Attributes

Companion
object
Supertypes
class Object
trait Matchable
class Any

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, STen), BatchStreamState, BatchStreamBuffers], validationBatchesOverEpoch: Option[TrainingLoopContext => BatchStream[(I, STen), 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])])]
Generated with