amend

facebookresearch · Jul 31, 2024 · 6c1eae1 · 6c1eae1
1 parent 371a00c
commit 6c1eae1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -99,6 +99,10 @@ def gru_mlp_sequence_config() -> ModelConfig:
                 mlp_num_cells=[],
                 mlp_activation_class=nn.Tanh,
                 mlp_layer_class=nn.Linear,
+                n_layers=1,
+                bias=True,
+                dropout=0,
+                compile=False,
             ),
             MlpConfig(num_cells=[4], activation_class=nn.Tanh, layer_class=nn.Linear),
         ],