update conftest

apache · Jul 18, 2020 · 827c785 · 827c785
1 parent b260042
commit 827c785
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/submarine-sdk/pysubmarine/tests/ml/pytorch/model/conftest.py b/submarine-sdk/pysubmarine/tests/ml/pytorch/model/conftest.py
@@ -68,7 +68,9 @@ def get_model_param(tmpdir):
                 "num_features": 117581,
                 "out_features": 1,
                 "embedding_dim": 16,
+                "attention_dim": 64,
                 "hidden_units": [400, 400],
+                "dropout_rate": 0.3, 
                 "dropout_rates": [0.2, 0.2]
             }
         },