summaryrefslogtreecommitdiff
path: root/training/conf/lr_schedulers/cosine_annealing.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-18 00:59:17 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-18 00:59:17 +0200
commit062eebf0d690365cf7d9f6019d147ea195cc3a63 (patch)
tree0e6c6ba1a4c23ed9b51787101d4f7b588830256f /training/conf/lr_schedulers/cosine_annealing.yaml
parent7dfa030557ad500a54126b8af40f6f0c42c5d620 (diff)
Update configs
Diffstat (limited to 'training/conf/lr_schedulers/cosine_annealing.yaml')
-rw-r--r--training/conf/lr_schedulers/cosine_annealing.yaml13
1 files changed, 6 insertions, 7 deletions
diff --git a/training/conf/lr_schedulers/cosine_annealing.yaml b/training/conf/lr_schedulers/cosine_annealing.yaml
index c53ee3a..e8364f0 100644
--- a/training/conf/lr_schedulers/cosine_annealing.yaml
+++ b/training/conf/lr_schedulers/cosine_annealing.yaml
@@ -1,8 +1,7 @@
-cosine_annealing:
- _target_: torch.optim.lr_scheduler.CosineAnnealingLR
- T_max: 256
- eta_min: 0.0
- last_epoch: -1
+_target_: torch.optim.lr_scheduler.CosineAnnealingLR
+T_max: 256
+eta_min: 0.0
+last_epoch: -1
- interval: epoch
- monitor: val/loss
+interval: epoch
+monitor: val/loss