From 2408aa886dbf93c9755cca05c037ebb62a930da5 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 18:00:15 -0400 Subject: [PATCH] testing hpc save load --- pytorch_lightning/root_module/model_saving.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 818c194748..1e47e3e751 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -144,7 +144,7 @@ class TrainerIO(object): filepath = '{}/hpc_ckpt_{}.ckpt'.format(folderpath, ckpt_number) # give model a chance to do something on hpc_save - self.on_hpc_save() + self.model.on_hpc_save() # request what to save from the model checkpoint_dict = self.dump_checkpoint() @@ -168,7 +168,7 @@ class TrainerIO(object): model.load_model_specific(checkpoint) # call model hook - self.on_hpc_load() + self.model.on_hpc_load() def max_ckpt_in_folder(self, path): files = os.listdir(path)