Skip to content

Commit 1a568fe

Browse files
Parameterize model device for demo and tests (#3875)
1 parent bb98fe4 commit 1a568fe

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

demos/action_recognition_demo/python/action_recognition_demo.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ def main():
9090
decoder_target_device = 'CPU'
9191
if args.device != 'CPU':
9292
encoder_target_device = args.device
93+
decoder_target_device = args.device
9394
else:
9495
encoder_target_device = decoder_target_device
9596

demos/tests/cases.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def single_option_cases(key, *args):
251251

252252
CppDemo(name='gaze_estimation_demo', implementation='cpp_gapi',
253253
model_keys=['-m', '-m_fd', '-m_hp', '-m_lm', '-m_es'],
254-
device_keys=['-d', '-d_fd', '-d_hp', '-d_lm'],
254+
device_keys=['-d', '-d_fd', '-d_hp', '-d_lm', '-d_es'],
255255
test_cases=combine_cases(
256256
TestCase(options={'-no_show': None,
257257
**MONITORS,
@@ -443,7 +443,7 @@ def single_option_cases(key, *args):
443443

444444
CppDemo(name='gaze_estimation_demo',
445445
model_keys=['-m', '-m_fd', '-m_hp', '-m_lm', '-m_es'],
446-
device_keys=['-d', '-d_fd', '-d_hp', '-d_lm'],
446+
device_keys=['-d', '-d_fd', '-d_hp', '-d_lm', '-d_es'],
447447
test_cases=combine_cases(
448448
TestCase(options={'-no_show': None,
449449
**MONITORS,
@@ -1239,7 +1239,7 @@ def single_option_cases(key, *args):
12391239
ModelArg('instance-segmentation-security-1040')),
12401240
)),
12411241

1242-
PythonDemo(name='machine_translation_demo', device_keys=[], test_cases=combine_cases(
1242+
PythonDemo(name='machine_translation_demo', device_keys=['-d'], test_cases=combine_cases(
12431243
[
12441244
TestCase(options={
12451245
'-m': ModelArg('machine-translation-nar-de-en-0002'),

0 commit comments

Comments
 (0)