From fede6ca1dd0077ff509d84bd24028cc7a93bb119 Mon Sep 17 00:00:00 2001
From: StevenLiuWen <liuwen@shanghaitech.edu.cn>
Date: Tue, 13 Mar 2018 03:28:06 -0400
Subject: first commit

---
 Codes/flownet2/src/training_schedules.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 Codes/flownet2/src/training_schedules.py

(limited to 'Codes/flownet2/src/training_schedules.py')

diff --git a/Codes/flownet2/src/training_schedules.py b/Codes/flownet2/src/training_schedules.py
new file mode 100644
index 0000000..4db5aab
--- /dev/null
+++ b/Codes/flownet2/src/training_schedules.py
@@ -0,0 +1,12 @@
+LONG_SCHEDULE = {
+    'step_values': [400000, 600000, 800000, 1000000],
+    'learning_rates': [0.0001, 0.00005, 0.000025, 0.0000125, 0.00000625],
+    'momentum': 0.9,
+    'momentum2': 0.999,
+    'weight_decay': 0.0004,
+    'max_iter': 1200000,
+}
+
+FINETUNE_SCHEDULE = {
+    # TODO: Finetune schedule
+}
-- 
cgit v1.2.3-70-g09d2