From af5c7219f64c5471fa431f7e4063581a3b2b6b8d Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Thu, 16 May 2019 12:11:04 +0300
Subject: [PATCH] Fix test_onnx.py
A wrong model was used
---
tests/test_onnx.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_onnx.py b/tests/test_onnx.py
index d509312..55ee15d 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'
--
GitLab