未验证 提交 33d8697f 编写于 作者: D duanyanhui 提交者: GitHub

[npu-tipc] Minimize coco datasets for npu tipc (#8226)

* change coco dataset for npu tipc

* update link
上级 ec37e666
......@@ -68,6 +68,22 @@ grep -n '.yml' $FILENAME | cut -d ":" -f 1 \
sed -i 's/aligned: True/aligned: False/g' "$sub_config_path"
done
done
# NPU lacks operators such as deformable_conv, depthwise_conv2d_transpose,
# which will affects ips. Here, we reduce the number of coco training sets
# for npu tipc bencnmark. This is a temporary hack.
# # TODO(duanyanhui): add vision ops for npu
train_img_num=`cat $REPO_ROOT_PATH/dataset/coco/annotations/instances_train2017.json | grep -o file_name | wc -l`
exp_num=8
if [ ${train_img_num} != ${exp_num} ];then
echo "Replace with npu tipc coco training annotations"
mv $REPO_ROOT_PATH/dataset/coco/annotations/instances_train2017.json $REPO_ROOT_PATH/dataset/coco/annotations/instances_train2017_bak.json
wget https://paddle-device.bj.bcebos.com/tipc/instances_train2017.json
mv instances_train2017.json $REPO_ROOT_PATH/dataset/coco/annotations/
rm -f instances_train2017.json
fi
# pass parameters to test_train_inference_python.sh
cmd="bash test_tipc/test_train_inference_python.sh ${FILENAME} $2"
echo $cmd
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册