From 257c722d7cdeb5ba0b329e8a5fd204ed073c36b9 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Fri, 12 Mar 2021 02:29:23 +0800
Subject: [PATCH] update

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

diff --git a/train.py b/train.py
index 9ff5dcb..4b0fd5e 100644
--- a/train.py
+++ b/train.py
@@ -150,8 +150,7 @@ def trainer(ops,f_log):
                 optimizer.zero_grad()
                 step += 1
 
-
-            torch.save(model_.state_dict(), ops.model_exp + '{}-model_epoch-{}.pth'.format(ops.model,epoch))
+            torch.save(model_.state_dict(), ops.model_exp + '{}-size-{}-model_epoch-{}.pth'.format(ops.model,ops.img_size[0],epoch))
 
     except Exception as e:
         print('Exception : ',e) # 打印异常
-- 
GitLab