feature-training/org.pointyware.disco.training.interactors

Package-level declarations

Types

data class NetworkTrainingState(val elapsedEpochs: Int = 0, val trainer: SequentialTrainer, val snapshot: Snapshot)

A Training Controller provides an interface to manage training of a model and exposes properties to observe the training state.

class TrainingControllerImpl(exerciseRepository: ExerciseRepository, trainingScope: CoroutineScope) : TrainingController

data class TrainingState(val isTraining: Boolean = false, val epochsRemaining: Int = 0, val epochsElapsed: Int = 0, val networks: List<NetworkTrainingState> = emptyList())

Models basic training state and provides a handle to the trainer and network.

val lossKey: ComputationKey<Float>