Update train.py

6aa7d7bb · Eric.Lee2021 · e072d048 · 6aa7d7bb
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

未找到文件。
--- 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: