Packages

o

lamp.data

DataParallel

object DataParallel

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. DataParallel
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Value Members

  1. def oneEpoch[I, M <: GenericModule[I, Variable], S, C](epochCount: Long, trainingCallback: TrainingCallback, mainModel: ModelWithOptimizer[I, M], trainBatches: BatchStream[(I, STen), S, C], logger: Option[Logger], learningRateScheduleFactor: Double, models: Seq[SupervisedModel[I, M]], accumulateGradientOverNBatches: Int): IO[Double]

    Updates main model in place, returns average training loss

  2. def validationOneEpoch[I, M <: GenericModule[I, Variable], S, C](models: Seq[SupervisedModel[I, M]], validationBatches: BatchStream[(I, STen), S, C], validationCallback: ValidationCallback, logger: Option[Logger], epochCount: Long): IO[Double]