From 1ef464984937c1de262e93ae393b4c1d5e575fa2 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Tue, 12 Oct 2021 11:51:05 +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 b7b4365..7b321ca 100644
--- a/train.py
+++ b/train.py
@@ -155,7 +155,7 @@ if __name__ == "__main__":
         if Cosine_lr:
             lr_scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=5, eta_min=1e-5)
         else:
-            lr_scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=1, gamma=0.92)
+            lr_scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=1, gamma=0.94)
 
         train_dataset   = YoloDataset(train_lines, input_shape, num_classes, mosaic=mosaic, train = True)
         val_dataset     = YoloDataset(val_lines, input_shape, num_classes, mosaic=False, train = False)
@@ -192,7 +192,7 @@ if __name__ == "__main__":
         if Cosine_lr:
             lr_scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=5, eta_min=1e-5)
         else:
-            lr_scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=1, gamma=0.92)
+            lr_scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=1, gamma=0.94)
 
         train_dataset   = YoloDataset(train_lines, input_shape, num_classes, mosaic=mosaic, train = True)
         val_dataset     = YoloDataset(val_lines, input_shape, num_classes, mosaic=False, train = False)
-- 
GitLab