Fix.

Lightning-AI · Aug 15, 2020 · 8aad173 · 8aad173
1 parent 7d36aac
commit 8aad173
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/base/model_template.py b/tests/base/model_template.py
@@ -126,6 +126,7 @@ def get_default_hparams(continue_training: bool = False, hpc_exp_number: int = 0
             hidden_dim=1000,
             b1=0.5,
             b2=0.999,
+            save_hparams=True
         )
 
         if continue_training: