diff --git a/train.py b/train.py
index 0ecea637ec4f1fd6fc77343169e2ce17dcab1ce7..d084ff597a6a22b7070565a2fb4003e9b99da91b 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: