未验证 提交 029edf86 编写于 作者: D duanyanhui 提交者: GitHub

update tipc scripts for time-consuming model on npu (#780)

上级 a99e5064
......@@ -24,6 +24,16 @@ sed -i "s/--device:gpu/--device:npu/g" $FILENAME
sed -i "s/--benchmark:True/--benchmark:False/g" $FILENAME
dataline=`cat $FILENAME`
# change total iters/epochs for npu/xpu to accelaration
modelname=$(echo $FILENAME | cut -d '/' -f3)
if [ $modelname == "Pix2pix" ]; then
sed -i "s/lite_train_lite_infer=10/lite_train_lite_infer=1/g" $FILENAME
fi
if [ $modelname == "edvr" ]; then
sed -i "s/lite_train_lite_infer=100/lite_train_lite_infer=10/g" $FILENAME
fi
# change gpu to npu in execution script
sed -i 's/\"gpu\"/\"npu\"/g' test_tipc/test_train_inference_python.sh
sed -i 's/--gpus/--npus/g' test_tipc/test_train_inference_python.sh
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册