diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 1ac91fd308..ba0c89293e 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -21,6 +21,9 @@ class ModelIO(object): """ raise NotImplementedError + # ------------------------- + # OPTIONAL HOOKS + # ------------------------- def on_hpc_save(self): """ Hook to do whatever you need right before Slurm manager saves the model