lamp-data/lamp.data/DataParallel

DataParallel

object DataParallel
class Object
trait Matchable
class Any

Value members

Concrete methods

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