Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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'] start_epoch = state['epoch']
load_model_state_dict(model.module, state['model'], load_model_state_dict(model.module, state['model'],
strict=args.load_state_strict) strict=args.load_state_strict)
if 'optimizer' in state: if 'optimizer' in state:
optimizer.load_state_dict(state['optimizer']) optimizer.load_state_dict(state['optimizer'])