Module burn::train

source ·

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
TrainOutput

Traits

LearnerCallback
TrainStep
ValidStep