[show_model_with_bounding] fix the problem also for model with conv l…

…ayer
leondavi · Jul 14, 2024 · 3a24fce · 3a24fce
1 parent 6b66212
commit 3a24fce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src_py/nerlPlanner/JsonElementWorkerDefinitions.py b/src_py/nerlPlanner/JsonElementWorkerDefinitions.py
@@ -75,7 +75,7 @@
 # Maps from layer type to the functionality of layer mapping 
 LayerTypeToFunctionalMap = OrderedDict([
     ("Scaling" , ScalingMethodMap),
-    ("CNN" , None),
+    ("Conv" , None),
     ("Perceptron" , ActivationFunctionsMap),
     ("Pooling" , PoolingMethodMap),
     ("Probabilistic" , ProbabilisticActivationFunctionMap),