Skip to content

Commit 77f10ec

Browse files
authored
removed faulty tests in resnet (neuralmagic#31)
Fixed pruned quant test. Removed unused test in resnet 50 imagenette
1 parent 61b7f8b commit 77f10ec

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/sparsezoo/models/classification/test_resnet.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -139,10 +139,10 @@ def test_resnet_50(
139139
True,
140140
"pytorch",
141141
"sparseml",
142-
"imagenet",
142+
"imagenet-augmented",
143143
None,
144144
"pruned_quant",
145-
"conservative",
145+
"aggressive",
146146
None,
147147
),
148148
(True, "tensorflow_v1", "sparseml", "imagenet", None, "base", "none", None),
@@ -157,7 +157,6 @@ def test_resnet_50(
157157
"conservative",
158158
None,
159159
),
160-
(True, "pytorch", "sparseml", "imagenette", None, "pruned", "moderate", None),
161160
(
162161
True,
163162
"pytorch",

0 commit comments

Comments
 (0)