Skip to main content

Module training

Module training 

Source
Expand description

Training infrastructure.

ModuleContents
learnerMain train() function, model save/load helpers
schedulerRsqrt LR scheduler with warm-up and cool-down

Modules§

learner
Training loop using the Burn Learner abstraction.
scheduler
Learning-rate schedulers.