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