From 197bb35e6ed1ed3d4207bd26232ff987031966fd Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 10 Sep 2020 14:18:28 +0800
Subject: [PATCH] Update train.py

---
 train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 93e4335..bd4c286 100644
--- a/train.py
+++ b/train.py
@@ -190,7 +190,7 @@ if __name__ == "__main__":
         lr = 1e-3
         Batch_size = 4
         Init_Epoch = 0
-        Freeze_Epoch = 25
+        Freeze_Epoch = 50
         
         optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
         if Cosine_lr:
@@ -226,8 +226,8 @@ if __name__ == "__main__":
     if True:
         lr = 1e-4
         Batch_size = 2
-        Freeze_Epoch = 25
-        Unfreeze_Epoch = 50
+        Freeze_Epoch = 50
+        Unfreeze_Epoch = 100
 
         optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
         if Cosine_lr:
-- 
GitLab