removed bad hook call

This commit is contained in:
William Falcon
2019-08-07 07:30:02 -04:00
parent 82d63a9677
commit 95ec072d1e
@@ -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 = {