Skip to content
This repository was archived by the owner on Nov 3, 2022. It is now read-only.

Commit fb46b9a

Browse files
RaphaelMeudectaehoonlee
authored andcommitted
fix selected module in mobilenet test : it should be dependant on selected mobilenet version (#25)
1 parent 8ad3241 commit fb46b9a

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/applications_test.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030
from multiprocessing import Process, Queue
3131

3232

33-
MOBILENET_LIST = [(mobilenet.MobileNet, 1024),
34-
(mobilenet_v2.MobileNetV2, 1280)]
33+
MOBILENET_LIST = [(mobilenet.MobileNet, mobilenet, 1024),
34+
(mobilenet_v2.MobileNetV2, mobilenet_v2, 1280)]
3535
DENSENET_LIST = [(densenet.DenseNet121, 1024),
3636
(densenet.DenseNet169, 1664),
3737
(densenet.DenseNet201, 1920)]
@@ -190,8 +190,7 @@ def test_inceptionresnetv2():
190190

191191

192192
def test_mobilenet():
193-
app, last_dim = random.choice(MOBILENET_LIST)
194-
module = mobilenet
193+
app, module, last_dim = random.choice(MOBILENET_LIST)
195194
_test_application_basic(app, module=module)
196195
_test_application_notop(app, last_dim)
197196
_test_application_variable_input_channels(app, last_dim)

0 commit comments

Comments
 (0)