diff options
Diffstat (limited to 'training/conf/callbacks/swa.yaml')
| -rw-r--r-- | training/conf/callbacks/swa.yaml | 7 |
1 files changed, 0 insertions, 7 deletions
diff --git a/training/conf/callbacks/swa.yaml b/training/conf/callbacks/swa.yaml deleted file mode 100644 index 73f8c66..0000000 --- a/training/conf/callbacks/swa.yaml +++ /dev/null @@ -1,7 +0,0 @@ -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 |