From 6c3c8538c5ae11062ffc12d9025de43f5cbc6116 Mon Sep 17 00:00:00 2001 From: anzhengqi Date: Wed, 22 Sep 2021 09:28:53 +0800 Subject: [PATCH] update retinaface_resnet50 network testcase scripts. --- .../retinaface_resnet50/test_retinaface_resnet50.py | 1 + 1 file changed, 1 insertion(+) 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"