From aacf1947ea056fa53122faa8dab2fd5843da7044 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 21:38:06 -0400 Subject: [PATCH] auto state-dict and remove the way the model is loaded during hpc --- pytorch_lightning/root_module/model_saving.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index ec55d5f62f..f2cb517ff5 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -4,6 +4,7 @@ import re import pdb from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel, LightningDataParallel + class ModelIO(object): def load_model_specific(self, checkpoint):