wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

added on_hpc_load and on_hpc_save hooks

Browse Source
This commit is contained in:
William Falcon
2019-07-02 09:36:33 -04:00
parent f257c080c0
commit 62e091f48d
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/model_saving.py
+3
View File
@@ -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