diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index d9c2e3a5..00b95460 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -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 = {