From 98c0dab13985cb74abb9d00e61462c7fe24fa78e Mon Sep 17 00:00:00 2001 From: Olivier DEBAUCHE Date: Fri, 26 Jul 2024 14:20:29 +0200 Subject: [PATCH] Update tests_models_mobilenet.py --- fl_common/tests/tests_models_mobilenet.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/fl_common/tests/tests_models_mobilenet.py b/fl_common/tests/tests_models_mobilenet.py index 8de25367..b65f79ae 100644 --- a/fl_common/tests/tests_models_mobilenet.py +++ b/fl_common/tests/tests_models_mobilenet.py @@ -55,11 +55,11 @@ def test_get_mobilenet_model(self): "mobilenetv4_conv_blur_large", "mobilenetv4_hybrid_medium_075", "mobilenetv4_hybrid_large_075", - "repghostnetv2_conv_small", - "repghostnetv2_conv_medium", - "repghostnetv2_conv_large", - "repghostnetv2_conv_aa_medium", - "repghostnetv2_conv_blur_medium", + # "repghostnetv2_conv_small", + # "repghostnetv2_conv_medium", + # "repghostnetv2_conv_large", + # cc"repghostnetv2_conv_aa_medium", + # "repghostnetv2_conv_blur_medium", ] num_classes = 10 # You can adjust the number of classes as needed