Commit 5cd956ef authored by mashun1's avatar mashun1
Browse files

Update train.py

parent f67df50a
......@@ -94,7 +94,7 @@ def train_pipeline(root_path):
opt['root_path'] = root_path
# torch.backends.cudnn.benchmark = True
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.deterministic = False
# load resume states if necessary
resume_state = load_resume_state(opt)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment