summaryrefslogtreecommitdiff
path: root/training/conf/lr_schedulers
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/lr_schedulers')
-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