Allow no ic_params on conv layers

sjawhar · Mar 18, 2020 · a13ced6 · a13ced6
1 parent 625978b
commit a13ced6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/app/src/models.py b/app/src/models.py
@@ -79,6 +79,8 @@ def add_conv1d_layer(model, name, ic_params={}, pool=None, **kwargs):
         elif type(pool) is int:
             pool = {PARAM_POOL_SIZE: pool}
         add_layer(model, MaxPooling1D, f"{name}_pool", ic_params=None, **pool)
+    if type(ic_params) is not dict:
+        return
     add_ic_layer(model, name, **ic_params)