auto state-dict and remove the way the model is loaded during hpc

2019-07-26 21:38:06 -04:00 · 2019-07-26 21:38:06 -04:00 · aacf1947ea
parent e2c7fa44b7
commit aacf1947ea
1 changed files with 1 additions and 0 deletions
--- 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):