removed bad hook call

This commit is contained in:
William Falcon
2019-08-07 07:39:41 -04:00
parent d5fd16a478
commit 5c398d7a4e
@@ -155,8 +155,6 @@ 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)