From e072d0482f11d828042f92b2b54f452cd6d35a68 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Fri, 23 Apr 2021 03:25:02 +0800
Subject: [PATCH] correct epoch random seed bug

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

diff --git a/train.py b/train.py
index 0ecea63..d084ff5 100644
--- a/train.py
+++ b/train.py
@@ -28,7 +28,7 @@ import cv2
 import time
 import json
 from datetime import datetime
-
+import random
 def trainer(ops,f_log):
     try:
         os.environ['CUDA_VISIBLE_DEVICES'] = ops.GPUS
@@ -164,6 +164,7 @@ def trainer(ops,f_log):
                 optimizer.zero_grad()
                 step += 1
 
+            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