From ae2a3c4a31871d35ed4dd9af920073c297836952 Mon Sep 17 00:00:00 2001
From: Ryan <wangzhiwei73@huawei.com>
Date: Fri, 25 Mar 2022 11:11:33 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0Voxel=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E8=84=9A=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../pose_estimation/VoxelPose/test/train_performance_8p.sh   | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/PyTorch/contrib/cv/pose_estimation/VoxelPose/test/train_performance_8p.sh b/PyTorch/contrib/cv/pose_estimation/VoxelPose/test/train_performance_8p.sh
index 86d1921766..0c2a718cde 100644
--- a/PyTorch/contrib/cv/pose_estimation/VoxelPose/test/train_performance_8p.sh
+++ b/PyTorch/contrib/cv/pose_estimation/VoxelPose/test/train_performance_8p.sh
@@ -62,6 +62,10 @@ etp_flag=`echo ${check_etp_flag#*=}`
 if [ x"${etp_flag}" != x"true" ];then
     source ${test_path_dir}/env_npu.sh
 fi
+
+for((RANK_ID=0;RANK_ID<RANK_SIZE;RANK_ID++));
+do
+export RANK_ID=$RANK_ID
 python3.7 run/train_3d.py \
     --cfg configs/shelf/prn64_cpn80x80x20.yaml \
     --data_path ${data_path} \
@@ -69,6 +73,7 @@ python3.7 run/train_3d.py \
     --num_epochs ${num_epochs} \
     --apex \
     --addr $(hostname -I |awk '{print $1}') > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 &
+done
 
 wait
 
-- 
Gitee