Update dlcl_getfamilly.py

Smartappli · Jul 26, 2024 · cef9b87 · cef9b87
1 parent 43e8e44
commit cef9b87
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fl_common/dlcl_getfamilly.py b/fl_common/dlcl_getfamilly.py
@@ -468,9 +468,9 @@ def get_family_model_e(model_type, num_classes):
         "semnasnet_100": get_efficientnet_model,
         "semnasnet_140": get_efficientnet_model,
         "mnasnet_small": get_efficientnet_model,
-        "mobilenet_100": get_efficientnet_model,
-        "mobilenet_100h": get_efficientnet_model,
-        "mobilenet_125": get_efficientnet_model,
+        "mobilenetv1_100": get_efficientnet_model,
+        "mobilenetv1_100h": get_efficientnet_model,
+        "mobilenetv1_125": get_efficientnet_model,
         "mobilenetv2_035": get_efficientnet_model,
         "mobilenetv2_050": get_efficientnet_model,
         "mobilenetv2_075": get_efficientnet_model,