From cef9b87137c34975817d07831e3b5f61024dc7c9 Mon Sep 17 00:00:00 2001 From: Olivier DEBAUCHE Date: Fri, 26 Jul 2024 14:02:40 +0200 Subject: [PATCH] Update dlcl_getfamilly.py --- fl_common/dlcl_getfamilly.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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,