diff --git a/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/test/train_full_8p.sh b/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/test/train_full_8p.sh index 31883a2fb41797d317e20b0a56f2ad3bf1e99555..d1bc9f06b6cb4347d09edba9a260d637339092db 100644 --- a/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/test/train_full_8p.sh +++ b/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/test/train_full_8p.sh @@ -10,7 +10,8 @@ if [ x"${cur_path_last_dirname}" == x"test" ];then else test_path_dir=${cur_path}/test fi - +#网络名称,同目录名称 +Network="Faster_Mask_RCNN_ID0101_for_PyTorch" # 指定训练所使用的npu device卡id device_id=0 batch_size=64 @@ -139,7 +140,7 @@ echo "E2E Training Duration sec : $e2e_time" #训练用例信息,不需要修改 BatchSize=${batch_size} DeviceType=`uname -m` -CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'perf' +CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'acc' ##获取性能数据 #吞吐量,不需要修改