From 62e091f48d8b20a730ab771d4b44c1049a704816 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 2 Jul 2019 09:36:33 -0400 Subject: [PATCH] added on_hpc_load and on_hpc_save hooks --- pytorch_lightning/root_module/model_saving.py | 3 +++ 1 file changed, 3 insertions(+) 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