diff options
Diffstat (limited to 'training/conf/lr_schedulers/one_cycle.yaml')
| -rw-r--r-- | training/conf/lr_schedulers/one_cycle.yaml | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/lr_schedulers/one_cycle.yaml b/training/conf/lr_schedulers/one_cycle.yaml index c60577a..801a01f 100644 --- a/training/conf/lr_schedulers/one_cycle.yaml +++ b/training/conf/lr_schedulers/one_cycle.yaml @@ -1,4 +1,4 @@ -onc_cycle: +one_cycle: _target_: torch.optim.lr_scheduler.OneCycleLR max_lr: 1.0e-3 total_steps: null |