diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-25 23:18:54 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-25 23:18:54 +0100 |
| commit | 05847d70dab99b14e0b9ee8102d403a0a4266535 (patch) | |
| tree | 85b5c3d17e31977cab8394d230799e305589519f /training | |
| parent | 1b7aa3665a720c627ef62a19076150ff449b0803 (diff) | |
Add config for reduce on plateau
Diffstat (limited to 'training')
| -rw-r--r-- | training/conf/lr_schedulers/reduce_on_plateau.yaml | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/training/conf/lr_schedulers/reduce_on_plateau.yaml b/training/conf/lr_schedulers/reduce_on_plateau.yaml new file mode 100644 index 0000000..9aa9de4 --- /dev/null +++ b/training/conf/lr_schedulers/reduce_on_plateau.yaml @@ -0,0 +1,12 @@ +_target_: torch.optim.lr_scheduler.ReduceLROnPlateau +mode: min +factor: 0.1 +patience: 10 +threshold: 1.0e-4 +threshold_mode: rel +cooldown: 0 +min_lr: 1.0e-5 +eps: 1.0e-8 +verbose: false +interval: epoch +monitor: val/loss |