diff --git a/tests/test_onnx.py b/tests/test_onnx.py index d509312dcab00f5b13b4eb86bdb8c60d3e01452d..55ee15d7e2c20b93f1439945437f96c3b2311763 100644 --- a/tests/test_onnx.py +++ b/tests/test_onnx.py @@ -31,7 +31,7 @@ logger.addHandler(fh) @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'