From fb46b9a63150374f49b271285c58d89f82f01ac3 Mon Sep 17 00:00:00 2001 From: Raphael Meudec Date: Fri, 3 Aug 2018 10:42:10 +0200 Subject: [PATCH] fix selected module in mobilenet test : it should be dependant on selected mobilenet version (#25) --- tests/applications_test.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/applications_test.py b/tests/applications_test.py index db03042..0f7f14a 100644 --- a/tests/applications_test.py +++ b/tests/applications_test.py @@ -30,8 +30,8 @@ from multiprocessing import Process, Queue -MOBILENET_LIST = [(mobilenet.MobileNet, 1024), - (mobilenet_v2.MobileNetV2, 1280)] +MOBILENET_LIST = [(mobilenet.MobileNet, mobilenet, 1024), + (mobilenet_v2.MobileNetV2, mobilenet_v2, 1280)] DENSENET_LIST = [(densenet.DenseNet121, 1024), (densenet.DenseNet169, 1664), (densenet.DenseNet201, 1920)] @@ -190,8 +190,7 @@ def test_inceptionresnetv2(): def test_mobilenet(): - app, last_dim = random.choice(MOBILENET_LIST) - module = mobilenet + app, module, last_dim = random.choice(MOBILENET_LIST) _test_application_basic(app, module=module) _test_application_notop(app, last_dim) _test_application_variable_input_channels(app, last_dim)