mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
removed bad hook call
This commit is contained in:
1 parent
82d63a9677
commit
95ec072d1e
1 file changed
-3
@@ -74,9 +74,6 @@ class TrainerIO(object):
|
||||
# load the state_dict on the model automatically
|
||||
model.load_state_dict(checkpoint['state_dict'])
|
||||
|
||||
# call model hook
|
||||
model.on_hpc_load(checkpoint)
|
||||
|
||||
def dump_checkpoint(self):
|
||||
|
||||
checkpoint = {
|
||||
|
||||
Reference in new issue
Block a user