mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-26 13:00:58 +08:00
removed bad hook call
This commit is contained in:
1 parent
d5fd16a478
commit
5c398d7a4e
1 file changed
-2
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user