diff --git a/train.py b/train.py
index 9ff5dcbf5298e593e7898a32b31fccf0ccd5d6c6..4b0fd5ece771f71e17ed916313a0faa88785206a 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) # 打印异常