From 9c75be5ce67f3918ecad5ba1cb3a65a06e141fc2 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Thu, 21 Jul 2022 14:10:14 +0800
Subject: [PATCH] Update train.py

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 26d185b..23504f1 100644
--- a/train.py
+++ b/train.py
@@ -21,10 +21,10 @@ log_path = 'train_log'
 def get_learning_rate(step):
     if step < 2000:
         mul = step / 2000.
-        return 1e-4 * mul
+        return 3e-4 * mul
     else:
         mul = np.cos((step - 2000) / (args.epoch * args.step_per_epoch - 2000.) * math.pi) * 0.5 + 0.5
-        return (1e-4 - 1e-5) * mul + 1e-5
+        return (3e-4 - 3e-6) * mul + 3e-6
 
 def flow2rgb(flow_map_np):
     h, w, _ = flow_map_np.shape