From 66ad83d64794cd03b0b95edb176f158190859070 Mon Sep 17 00:00:00 2001
From: James Yu <30967260+jbrightuniverse@users.noreply.github.com>
Date: Mon, 22 Nov 2021 10:38:47 -0800
Subject: [PATCH] defaults for flexiblesequential

---
 econ_layers/layers.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/econ_layers/layers.py b/econ_layers/layers.py
index a1e775e..8efccd3 100644
--- 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),