removed bad hook call

This commit is contained in:
William Falcon 2019-08-07 07:35:02 -04:00
parent 9713c41bf4
commit a931ded310
1 changed files with 2 additions and 0 deletions

View File

@ -155,6 +155,8 @@ class TrainerIO(object):
self.current_epoch = checkpoint['epoch']
# restore the optimizers
import pdb
pdb.set_trace()
optimizer_states = checkpoint['optimizer_states']
for optimizer, opt_state in zip(self.optimizers, optimizer_states):
optimizer.load_state_dict(opt_state)