Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix indentation bug pointed out by Patrick Lachance

Browse Source
This commit is contained in:
Yin Li 2020-09-21 19:54:44 -04:00
parent 7971137ad4
commit 156011be5f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/train.py
View File

@ -157,7 +157,7 @@ def gpu_worker(local_rank, node, args):
start_epoch = state['epoch']
load_model_state_dict(model.module, state['model'],
strict=args.load_state_strict)
strict=args.load_state_strict)
if 'optimizer' in state:
optimizer.load_state_dict(state['optimizer'])