summaryrefslogtreecommitdiff
path: root/training/conf/lr_scheduler
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/lr_scheduler')
-rw-r--r--training/conf/lr_scheduler/one_cycle.yaml8
1 files changed, 8 insertions, 0 deletions
diff --git a/training/conf/lr_scheduler/one_cycle.yaml b/training/conf/lr_scheduler/one_cycle.yaml
new file mode 100644
index 0000000..60a6f27
--- /dev/null
+++ b/training/conf/lr_scheduler/one_cycle.yaml
@@ -0,0 +1,8 @@
+type: OneCycleLR
+args:
+ interval: step
+ max_lr: 1.0e-3
+ three_phase: true
+ epochs: 64
+ steps_per_epoch: 633 # num_samples / batch_size
+monitor: val_loss