training/conf/callbacks/swa.yaml


1
2
3
4
5
6
7
8

stochastic_weight_averaging:
  type: StochasticWeightAveraging
  args:
    swa_epoch_start: 0.8
    swa_lrs: 0.05
    annealing_epochs: 10
    annealing_strategy: cos
    device: null