Update tests_models_mobilenet.py

Smartappli · Jul 26, 2024 · 98c0dab · 98c0dab
1 parent 518778c
commit 98c0dab
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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