summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/swa.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-10 18:06:54 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-10 18:06:54 +0200
commitb69254ce3135c112e29f7f1c986b7f0817da0c33 (patch)
treee596a719b445c0cbbd4108079206ec9d14de1437 /training/conf/callbacks/swa.yaml
parentcba94bdcab90f288dd1172607500ba2b28279736 (diff)
Update configs
Diffstat (limited to 'training/conf/callbacks/swa.yaml')
-rw-r--r--training/conf/callbacks/swa.yaml7
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