Crate burn_train

Modules

• checkpoint
• logger
• metric
• train

Structs

• AsyncTrainerCallback
• ClassificationOutput
  Simple classification output adapted for multiple metrics.
• Learner
  Learner struct encapsulating all components necessary to train a Neural Network model.
• LearnerBuilder
  Struct to configure and create a learner.
• LearnerItem
• MultiDevicesTrainStep
• TrainEpoch
• TrainOutput
• ValidEpoch

Traits

• LearnerCallback
• TrainStep
• ValidStep