From 2c9066b685d41ef0ab5ea94e938b8a30b4123656 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 15 Jun 2022 00:15:40 +0200
Subject: Update configs

---
 training/conf/lr_schedulers/one_cycle.yaml | 37 ++++++++++++++----------------
 1 file changed, 17 insertions(+), 20 deletions(-)

(limited to 'training/conf/lr_schedulers')

diff --git a/training/conf/lr_schedulers/one_cycle.yaml b/training/conf/lr_schedulers/one_cycle.yaml
index 801a01f..20eab9f 100644
--- a/training/conf/lr_schedulers/one_cycle.yaml
+++ b/training/conf/lr_schedulers/one_cycle.yaml
@@ -1,20 +1,17 @@
-one_cycle:
-  _target_: torch.optim.lr_scheduler.OneCycleLR
-  max_lr: 1.0e-3
-  total_steps: null
-  epochs: 512
-  steps_per_epoch: 4992
-  pct_start: 0.3
-  anneal_strategy: cos
-  cycle_momentum: true
-  base_momentum: 0.85
-  max_momentum: 0.95
-  div_factor: 25.0
-  final_div_factor: 10000.0
-  three_phase: true
-  last_epoch: -1
-  verbose: false
-
-  # Non-class arguments
-  interval: step
-  monitor: val/loss
+_target_: torch.optim.lr_scheduler.OneCycleLR
+max_lr: 1.0e-3
+total_steps: null
+epochs: 512
+steps_per_epoch: 4992
+pct_start: 0.3
+anneal_strategy: cos
+cycle_momentum: true
+base_momentum: 0.85
+max_momentum: 0.95
+div_factor: 25.0
+final_div_factor: 10000.0
+three_phase: true
+last_epoch: -1
+verbose: false
+interval: step
+monitor: val/loss
-- 
cgit v1.2.3-70-g09d2