defaults for flexiblesequential

HighDimensionalEconLab · Nov 22, 2021 · 66ad83d · 66ad83d
1 parent 42d5d7e
commit 66ad83d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/econ_layers/layers.py b/econ_layers/layers.py
@@ -57,10 +57,10 @@ def forward(self, input):
 class FlexibleSequential(nn.Module):
     def __init__(
         self,
-        n_in: int,
-        n_out: int,
-        layers: int,
-        hidden_dim: int,
+        n_in: int = 1,
+        n_out: int = 1,
+        layers: int = 4,
+        hidden_dim: int = 128,
         activator: Optional[nn.Module] = lazy_instance(nn.ReLU),
         hidden_bias: bool = True,
         last_activator: Optional[nn.Module] = lazy_instance(nn.Identity),