From 6aa7d7bb8af9a3f4958abbca79980a9929bde50f Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Wed, 28 Apr 2021 01:11:34 +0800
Subject: [PATCH] Update train.py

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

diff --git a/train.py b/train.py
index d084ff5..b4f9de7 100644
--- a/train.py
+++ b/train.py
@@ -164,7 +164,7 @@ def trainer(ops,f_log):
                 optimizer.zero_grad()
                 step += 1
 
-            set_seed(random.randint(0,65535))
+            #set_seed(random.randint(0,65535))
             torch.save(model_.state_dict(), ops.model_exp + '{}-size-{}-loss-{}-model_epoch-{}.pth'.format(ops.model,ops.img_size[0],ops.loss_define,epoch))
 
     except Exception as e:
-- 
GitLab