diff --git a/fl_common/dlcl_getfamilly.py b/fl_common/dlcl_getfamilly.py index 7f0c30ca..d7d7338c 100644 --- a/fl_common/dlcl_getfamilly.py +++ b/fl_common/dlcl_getfamilly.py @@ -469,6 +469,7 @@ def get_family_model_e(model_type, num_classes): "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, "mobilenetv2_035": get_efficientnet_model, "mobilenetv2_050": get_efficientnet_model, @@ -950,6 +951,8 @@ def get_family_model_m(model_type, num_classes): "mobilenetv4_hybrid_large": get_mobilenet_model, "mobilenetv4_conv_aa_medium": get_mobilenet_model, "mobilenetv4_conv_blur_medium": get_mobilenet_model, + "mobilenetv4_conv_aa_large": get_mobilenet_model, + "mobilenetv4_conv_blur_large": get_mobilenet_model, "mobilenetv4_hybrid_medium_075": get_mobilenet_model, "mobilenetv4_hybrid_large_075": get_mobilenet_model, "repghostnetv2_conv_small": get_mobilenet_model, diff --git a/fl_common/models/efficientnet.py b/fl_common/models/efficientnet.py index 5a6636b4..f1e9eba6 100644 --- a/fl_common/models/efficientnet.py +++ b/fl_common/models/efficientnet.py @@ -77,7 +77,7 @@ def get_efficientnet_model(efficientnet_type, num_classes): "semnasnet_140", "mnasnet_small", "mobilenet_100", - # "mobilenet_100h", + "mobilenet_100h", "mobilenet_125", "mobilenetv2_035", "mobilenetv2_050", diff --git a/fl_common/models/mobilenet.py b/fl_common/models/mobilenet.py index 5393c9e5..07d15d02 100644 --- a/fl_common/models/mobilenet.py +++ b/fl_common/models/mobilenet.py @@ -69,6 +69,8 @@ def get_mobilenet_model(mobilenet_type, num_classes): "mobilenetv4_hybrid_large", "mobilenetv4_conv_aa_medium", "mobilenetv4_conv_blur_medium", + "mobilenetv4_conv_aa_large", + "mobilenetv4_conv_blur_large", "mobilenetv4_hybrid_medium_075", "mobilenetv4_hybrid_large_075", "repghostnetv2_conv_small", diff --git a/fl_common/tests/tests_models_efficientnet.py b/fl_common/tests/tests_models_efficientnet.py index 47265202..e9cbd307 100644 --- a/fl_common/tests/tests_models_efficientnet.py +++ b/fl_common/tests/tests_models_efficientnet.py @@ -42,6 +42,7 @@ def test_efficientnet_model(self): "semnasnet_140", "mnasnet_small", "mobilenet_100", + # "mobilenet_100h", "mobilenet_125", "mobilenetv2_035", "mobilenetv2_050", diff --git a/fl_common/tests/tests_models_mobilenet.py b/fl_common/tests/tests_models_mobilenet.py index a491e6a8..44d02d1c 100644 --- a/fl_common/tests/tests_models_mobilenet.py +++ b/fl_common/tests/tests_models_mobilenet.py @@ -51,6 +51,8 @@ def test_get_mobilenet_model(self): "mobilenetv4_hybrid_large", "mobilenetv4_conv_aa_medium", "mobilenetv4_conv_blur_medium", + # "mobilenetv4_conv_aa_large", + # "mobilenetv4_conv_blur_large", "mobilenetv4_hybrid_medium_075", "mobilenetv4_hybrid_large_075", # "repghostnetv2_conv_small",