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

DataParallel

lamp.data.DataParallel
object DataParallel

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Self type

Members list

Value members

Concrete methods

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

Updates main model in place, returns average training loss

Attributes

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]
Generated with