diff --git a/fl_common/dlcl_getfamilly.py b/fl_common/dlcl_getfamilly.py index d7d7338c..260ccd24 100644 --- 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,