Skip to content

Stochastic Weight Averaging

Category: callback.swa
Version: 1.0.0
UUID: e6c5ffa9f27133aebb4398ce44333ae8ec2a808e38794bdc527b497311911b7f
Execution context: leip_af
Choice Priority: 400.0

Value Parameters

Name Synonyms Type Values
# of epochs for annealing swa.annealing_epochs scalar int
Annealing strategy swa.annealing_strategy choice cos
linear
Start epoch for SWA swa.epoch_start scalar int
SWA Learning rate swa.lr scalar float

This component fits into

Name UUID Synonyms
Full Recipe 8b4b05... callbacks