From 27cb95f1c4451ddea5390a08c78607a753e70af8 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Sun, 24 Jan 2021 11:15:56 +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 fb3f156..4449c27 100644
--- a/train.py
+++ b/train.py
@@ -191,7 +191,7 @@ if __name__ == "__main__":
         Freeze_Epoch = 50
         
         optimizer = optim.Adam(net.parameters(),lr)
-        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.95)
+        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.92)
 
         if Use_Data_Loader:
             train_dataset = YoloDataset(lines[:num_train], (Config["img_h"], Config["img_w"]), True)
@@ -225,7 +225,7 @@ if __name__ == "__main__":
         Unfreeze_Epoch = 100
 
         optimizer = optim.Adam(net.parameters(),lr)
-        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.95)
+        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.92)
         
         if Use_Data_Loader:
             train_dataset = YoloDataset(lines[:num_train], (Config["img_h"], Config["img_w"]), True)
-- 
GitLab