fixed hpc save, load. cleaned apu

2019-07-26 22:09:35 -04:00 · 2019-07-26 22:09:35 -04:00 · 348223a702
parent 64de447545
commit 348223a702
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/root_module/model_saving.py
+++ b/pytorch_lightning/root_module/model_saving.py
@ -79,7 +79,7 @@ class TrainerIO(object):

        # add the state_dict from the model
        model = self.__get_model()
-        checkpoint['state_dict'] = model.get_state_dict
+        checkpoint['state_dict'] = model.state_dict()

        # give the model a chance to add a few things
        model.on_save_checkpoint(checkpoint)