update

c9996883 · Eric.Lee2021 · 082fbc96 · c9996883
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

未找到文件。
--- 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. # 损失均值