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