From 9529aa6cc8dc458b01a97a5f6cf0461a7f197000 Mon Sep 17 00:00:00 2001 From: NicEggert Date: Tue, 22 Oct 2019 16:21:19 -0500 Subject: [PATCH] Add warning when not saving hparams --- pytorch_lightning/trainer/trainer_io.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer_io.py b/pytorch_lightning/trainer/trainer_io.py index 30afeddc..98bf58a9 100644 --- a/pytorch_lightning/trainer/trainer_io.py +++ b/pytorch_lightning/trainer/trainer_io.py @@ -175,7 +175,13 @@ class TrainerIOMixin(object): # add the hparams and state_dict from the model model = self.get_model() checkpoint['state_dict'] = model.state_dict() - checkpoint['hparams'] = vars(model.hparams) + if hasattr(model, "hparams"): + checkpoint['hparams'] = vars(model.hparams) + else: + warnings.warn( + "Did not find hyperparameters at model.hparams. Saving checkpoint without" + " hyperparameters" + ) # give the model a chance to add a few things model.on_save_checkpoint(checkpoint)