Fix test_onnx.py

A wrong model was used
chenxiang204 · May 16, 2019 · af5c721 · af5c721
1 parent 2ef3eeb
commit af5c721
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_onnx.py b/tests/test_onnx.py
@@ -31,7 +31,7 @@
 
 
 @pytest.mark.parametrize('arch',
-    ['resnet18', 'resnet20_cifar', 'alexnet', 'vgg19', 'resnext50_32x4d'])
+    ['resnet18', 'resnet20_cifar', 'alexnet', 'vgg19', 'resnext101_32x4d'])
 @pytest.mark.parametrize('add_softmax', [True, False])
 def test_summary(arch, add_softmax):
     dataset = 'cifar10' if arch.endswith('cifar') else 'imagenet'