From 348223a702bfe4724e9ec8ac46050e46b445eba5 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 22:09:35 -0400 Subject: [PATCH] fixed hpc save, load. cleaned apu --- pytorch_lightning/root_module/model_saving.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 03cb5864a8..142d2b337b 100644 --- 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)