fixed missing arg in evalmodel

Lightning-AI · Aug 15, 2020 · 2d9cf69 · 2d9cf69
1 parent 26aac08
commit 2d9cf69
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/base/model_template.py b/tests/base/model_template.py
@@ -64,6 +64,7 @@ def __init__(
         self.hidden_dim = hidden_dim
         self.b1 = b1
         self.b2 = b2
+        self.save_hparams = save_hparams
         self.training_step_called = False
         self.training_step_end_called = False
         self.training_epoch_end_called = False