!49608 fix deeplabv3 and ssd_mobilenet_fpn testcase

Merge pull request !49608 from anzhengqi/fix-ssd
This commit is contained in:
i-robot 2023-03-02 02:38:35 +00:00 committed by Gitee
commit a45992107d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions

View File

@ -46,7 +46,7 @@ def test_DeeplabV3_voc2007():
model_name, utils.rank_table_path)
ret = os.system(exec_network_shell)
assert ret == 0
cmd = "ps -ef | grep python | grep train.py | grep -v grep"
cmd = "ps -ef --columns 1000 | grep python | grep train.py | grep -v grep"
ret = utils.process_check(100, cmd)
assert ret

View File

@ -39,7 +39,7 @@ def test_SSD_mobilenet_v1_fpn_coco2017():
exec_network_shell = "cd {0}; sh -x scripts/run_distribute_train.sh 8 {1} 0.2 coco \
{2} config/ssd_mobilenet_v1_fpn_config.yaml".format(model_name, 60, utils.rank_table_path)
os.system(exec_network_shell)
cmd = "ps -ef | grep train.py | grep coco | grep device_num | grep device_id | grep -v grep"
cmd = "ps -ef --columns 1000 | grep train.py | grep coco | grep device_num | grep device_id | grep -v grep"
ret = utils.process_check(120, cmd)
assert ret