summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/swa.yaml
blob: 73f8c66b8d477985554211e10385b91d5c6c1ea4 (plain)
1
2
3
4
5
6
7
stochastic_weight_averaging:
  _target_: pytorch_lightning.callbacks.StochasticWeightAveraging
  swa_epoch_start: 0.8
  swa_lrs: 0.05
  annealing_epochs: 10
  annealing_strategy: cos
  device: null