diff --git a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_eval_8p.sh b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_eval_8p.sh index 20a9057a2a7d94d564284bcea2220ace674d7a9e..8fd04eb28b37f9dffb1efb5c27d87e605f78cffa 100644 --- a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_eval_8p.sh +++ b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_eval_8p.sh @@ -65,7 +65,7 @@ start_time=$(date +%s) check_etp_flag=`env | grep etp_running_flag` etp_flag=`echo ${check_etp_flag#*=}` if [ x"${etp_flag}" != x"true" ];then - source ../test/env.sh + source ../test/env_npu.sh fi nohup ./tools/dist_test.sh ./configs/fsaf/fsaf_r50_fpn_1x_coco.py $MODEL 8 --eval bbox > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & diff --git a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_1p.sh b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_1p.sh index f4e38919d2026cd877273e9f8bbc4c8da50f26ec..bee765e9fa15165482185c20de6ac7afd7a7af61 100644 --- a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_1p.sh +++ b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_1p.sh @@ -77,7 +77,7 @@ start_time=$(date +%s) check_etp_flag=`env | grep etp_running_flag` etp_flag=`echo ${check_etp_flag#*=}` if [ x"${etp_flag}" != x"true" ];then - source ../test/env.sh + source ../test/env_npu.sh fi python3.7 ./tools/train.py ./configs/fsaf/fsaf_r50_fpn_1x_coco.py > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & diff --git a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_8p.sh b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_8p.sh index 8407469f22529b3865d054ce43ef2540735073f6..47fce779e66645b545dec1660fa6a2c186365aae 100644 --- a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_8p.sh +++ b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_full_8p.sh @@ -62,7 +62,7 @@ start_time=$(date +%s) check_etp_flag=`env | grep etp_running_flag` etp_flag=`echo ${check_etp_flag#*=}` if [ x"${etp_flag}" != x"true" ];then - source ../test/env.sh + source ../test/env_npu.sh fi python3.7 ./tools/train.py ./configs/fsaf/fsaf_r50_fpn_1x_coco.py 8 > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & diff --git a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_1p.sh b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_1p.sh index 901d6aa49792fade5f2a203552f6e463d8c7e3dd..3ecd573e7af3ab6fb59d686057b02c92b3dfd0d7 100644 --- a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_1p.sh +++ b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_1p.sh @@ -76,7 +76,7 @@ start_time=$(date +%s) check_etp_flag=`env | grep etp_running_flag` etp_flag=`echo ${check_etp_flag#*=}` if [ x"${etp_flag}" != x"true" ];then - source ../test/env.sh + source ../test/env_npu.sh fi python3.7 ./tools/train.py ./configs/fsaf/fsaf_r50_fpn_1x_coco.py > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & diff --git a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_8p.sh b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_8p.sh index caac80274ff7e3689d34c0ba8740345286c7b2ba..8b0faa1eceb329090caebf7af51373f2306952ed 100644 --- a/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_8p.sh +++ b/PyTorch/contrib/cv/detection/FSAF_for_Pytorch/test/train_performance_8p.sh @@ -66,7 +66,7 @@ export TASK_QUEUE_ENABLE=0 check_etp_flag=`env | grep etp_running_flag` etp_flag=`echo ${check_etp_flag#*=}` if [ x"${etp_flag}" != x"true" ];then - source ../test/env.sh + source ../test/env_npu.sh fi python3.7 ./tools/train.py ./configs/fsaf/fsaf_r50_fpn_1x_coco.py 8 > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 &