diff --git a/tests/st/model_zoo_tests/retinaface_resnet50/test_retinaface_resnet50.py b/tests/st/model_zoo_tests/retinaface_resnet50/test_retinaface_resnet50.py index b46894a30c2..ef7187c13dd 100644 --- a/tests/st/model_zoo_tests/retinaface_resnet50/test_retinaface_resnet50.py +++ b/tests/st/model_zoo_tests/retinaface_resnet50/test_retinaface_resnet50.py @@ -51,6 +51,7 @@ def test_retinaface_resnet50(): old_list = ["python train.py"] new_list = ["python train.py --distributed 1 --device_target GPU"] utils.exec_sed_command(old_list, new_list, os.path.join(cur_model_path, "scripts/run_distribute_gpu_train.sh")) + os.system("nvidia-smi") exec_network_shell = "cd retinaface_resnet50; bash scripts/run_distribute_gpu_train.sh 4 1,2,3,4" os.system(exec_network_shell) cmd = "ps -ef | grep train.py | grep distributed | grep device_target | grep -v grep"