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