Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!23840 update retinaface_resnet50 network testcase scripts. 

Merge pull request !23840 from anzhengqi/update-testcase-retinaface
This commit is contained in:
i-robot 2021-09-24 09:39:37 +00:00 committed by Gitee
parent ee38ffbd3d 6c3c8538c5
commit d67c91edf5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/st/model_zoo_tests/retinaface_resnet50/test_retinaface_resnet50.py
View File

@ -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"