From c99968839628691d4b1597aebb2eb246e17dd438 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Thu, 25 Feb 2021 10:51:04 +0800
Subject: [PATCH] update

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 3cc4503..8c71b8b 100644
--- a/train.py
+++ b/train.py
@@ -101,7 +101,7 @@ def train(fintune_model,image_size,lr0,path_data,model_exp):
 
                 if flag_change_lr_cnt > 30:
                     init_lr = init_lr*0.1
-                    set_learning_rate(optimizer, init_lr)
+                    set_learning_rate(optim, init_lr)
                     flag_change_lr_cnt = 0
 
         loss_mean = 0. # 损失均值
-- 
GitLab