fix model selection

2017-10-09 15:10:15 +02:00
parent a686f147f0
commit d1da3d6ca3
3 changed files with 7 additions and 1 deletions
--- a/models/init.py
+++ b/models/init.py
@@ -29,6 +29,8 @@ def get_models_by_params(params: dict):
    elif network_depth == "flat2":
        networks = flat_2
    elif network_depth == "deep1":
+        networks = deep1
+    elif network_depth == "deep2":
        networks = renes_networks
    else:
        raise Exception("network not found")