From cd931c8220804fe399f69997455b14dea6a2ca44 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 11:40:45 -0400 Subject: [PATCH] added safeguards for callbacks in loading saving --- tests/debug.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/debug.py b/tests/debug.py index d3e478a47e..a713556817 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -25,7 +25,7 @@ def get_model(): 'hidden_dim': 1000}) model = LightningTemplateModel(hparams) - return model + return model, hparams def get_exp(debug=True): @@ -57,10 +57,11 @@ def clear_save_dir(): def main(): save_dir = init_save_dir() - model = get_model() + model, hparams = get_model() # exp file to get meta exp = get_exp(False) + exp.add_meta_from_hyperopt(hparams) exp.save() # exp file to get weights