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

update retinaface_resnet50 network testcase scripts.

This commit is contained in:
anzhengqi 2021-09-22 09:28:53 +08:00
parent c60e900000
commit 6c3c8538c5
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"