From 4da7a2c812221d56a430b35139ac40b23fa76f77 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 29 Jun 2021 22:54:52 +0200
Subject: Refactor of config, more granular

---
 training/conf/lr_scheduler/one_cycle.yaml | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 training/conf/lr_scheduler/one_cycle.yaml

(limited to 'training/conf/lr_scheduler/one_cycle.yaml')

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
-- 
cgit v1.2.3-70-g09d2