From 8652bb60fdae55118d832a783e75ad566a921f76 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Thu, 25 Feb 2021 11:02:06 +0800
Subject: [PATCH] update

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 8c71b8b..a619856 100644
--- a/train.py
+++ b/train.py
@@ -129,8 +129,8 @@ def train(fintune_model,image_size,lr0,path_data,model_exp):
 
             if it % msg_iter == 0:
 
-                print('epoch <{}/{}> -->> <{}/{}> -> iter {} : loss {:.5f}, loss_mean :{:.5f}, best_loss :{:.5f},lr :{:.6f},batch_size : {}'.\
-                format(epoch,max_epoch,i,int(ds.__len__()/n_img_per_gpu),it,loss.item(),loss_mean/loss_idx,best_loss,init_lr,n_img_per_gpu))
+                print('epoch <{}/{}> -->> <{}/{}> -> iter {} : loss {:.5f}, loss_mean :{:.5f}, best_loss :{:.5f},lr :{:.6f},batch_size : {},img_size :{}'.\
+                format(epoch,max_epoch,i,int(ds.__len__()/n_img_per_gpu),it,loss.item(),loss_mean/loss_idx,best_loss,init_lr,n_img_per_gpu,image_size))
 
                 if (it) % 500 == 0:
                     state = net.module.state_dict() if hasattr(net, 'module') else net.state_dict()
-- 
GitLab