From 454576954c9392aa739b4d8f53731b76aff0f684 Mon Sep 17 00:00:00 2001
From: gengdongjie <gengdongjie@huawei.com>
Date: Mon, 28 Dec 2020 09:36:57 +0800
Subject: [PATCH] fix maskrcnn 16p performance bug

---
 model_zoo/official/cv/maskrcnn/scripts/run_distribute_train.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_zoo/official/cv/maskrcnn/scripts/run_distribute_train.sh b/model_zoo/official/cv/maskrcnn/scripts/run_distribute_train.sh
index 77dc09dff3b..83b62cc9bee 100644
--- a/model_zoo/official/cv/maskrcnn/scripts/run_distribute_train.sh
+++ b/model_zoo/official/cv/maskrcnn/scripts/run_distribute_train.sh
@@ -54,7 +54,7 @@ export RANK_TABLE_FILE=$PATH1
 echo 3 > /proc/sys/vm/drop_caches
 
 cpus=`cat /proc/cpuinfo| grep "processor"| wc -l`
-avg=`expr $cpus \/ $RANK_SIZE`
+avg=`expr $cpus \/ $DEVICE_NUM`
 gap=`expr $avg \- 1`
 
 for((i=0; i<${DEVICE_NUM}; i++))