fix missing parameters, add flat network structure, make larger graphics

models/__init__.py

@@ -1,7 +1,6 @@
 import keras.backend as K
 
-from . import pauls_networks
-from . import renes_networks
+from . import flat_2, pauls_networks, renes_networks
 
 
 def get_models_by_params(params: dict):
@@ -27,6 +26,8 @@ def get_models_by_params(params: dict):
     # create models
     if network_depth == "small":
         networks = pauls_networks
+    elif network_depth == "flat":
+        networks = flat_2
     elif network_depth == "medium":
         networks = renes_networks
     else: