Skip to content

GitLab

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

Update train.py

parent f67df50a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
sr_model/Basicsr/basicsr/train.py
View file @ 5cd956ef
...@@ -94,7 +94,7 @@ def train_pipeline(root_path): ...@@ -94,7 +94,7 @@ def train_pipeline(root_path):
opt['root_path'] = root_path opt['root_path'] = root_path
# torch.backends.cudnn.benchmark = True # torch.backends.cudnn.benchmark = True
torch.backends.cudnn.deterministic = True torch.backends.cudnn.deterministic = False
# load resume states if necessary # load resume states if necessary
resume_state = load_resume_state(opt) 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