diff --git a/app/src/models.py b/app/src/models.py index 436ce6a..cd46e77 100644 --- 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)