Stub fixes (#1726)

* pass tests/sparseml/pytorch/models/classification/test_inception_v3.py * pass zoo:cv/classification/vgg-16_bn/pytorch/sparseml/imagenet/base-none * pass tests * track models, deprecated/corrupted * re-sync models from source, and lint
neuralmagic · Sep 25, 2023 · adb0eca · adb0eca
1 parent 59193ec
commit adb0eca
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 5 deletions.
diff --git a/tests/sparseml/pytorch/models/classification/test_inception_v3.py b/tests/sparseml/pytorch/models/classification/test_inception_v3.py
@@ -35,7 +35,7 @@
     [
         ("inception_v3", False, True),
         ("inception_v3", True, False),
-        ("inception_v3", "pruned-conservative", False),
+        ("inception_v3", "pruned-moderate", False),
     ],
 )
 def test_inception_v3s(key: str, pretrained: Union[bool, str], test_input: bool):

diff --git a/tests/sparseml/pytorch/models/classification/test_mobilenet.py b/tests/sparseml/pytorch/models/classification/test_mobilenet.py
@@ -36,7 +36,6 @@
         ("mobilenet", False, True),
         ("mobilenet", True, False),
         ("mobilenet", "base", False),
-        ("mobilenet", "pruned-conservative", False),
         ("mobilenet", "pruned-moderate", False),
     ],
 )

diff --git a/tests/sparseml/pytorch/models/classification/test_resnet.py b/tests/sparseml/pytorch/models/classification/test_resnet.py
@@ -63,8 +63,6 @@
         ("resnet50", False, True, resnet50),
         ("resnet50", True, False, resnet50),
         ("resnet50", "base", False, resnet50),
-        ("resnet50", "pruned-conservative", False, resnet50),
-        ("resnet50", "pruned-moderate", False, resnet50),
         ("resnet50_2xwidth", False, True, resnet50_2xwidth),
         ("resnet50_2xwidth", True, False, resnet50_2xwidth),
         ("resnet50_2xwidth", "base", False, resnet50_2xwidth),

diff --git a/tests/sparseml/pytorch/models/classification/test_vgg.py b/tests/sparseml/pytorch/models/classification/test_vgg.py
@@ -54,7 +54,6 @@
         ("vgg16", False, True, vgg16),
         ("vgg16", True, False, vgg16),
         ("vgg16", "base", False, vgg16),
-        ("vgg16", "pruned-conservative", False, vgg16),
         ("vgg16", "pruned-moderate", False, vgg16),
         ("vgg16bn", False, True, vgg16bn),
         ("vgg16bn", True, False, vgg16bn),