Skip to content

Commit 029edf8

Browse files
authored
update tipc scripts for time-consuming model on npu (#780)
1 parent a99e506 commit 029edf8

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

test_tipc/test_train_inference_python_npu.sh

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,16 @@ sed -i "s/--device:gpu/--device:npu/g" $FILENAME
2424
sed -i "s/--benchmark:True/--benchmark:False/g" $FILENAME
2525
dataline=`cat $FILENAME`
2626

27+
# change total iters/epochs for npu/xpu to accelaration
28+
modelname=$(echo $FILENAME | cut -d '/' -f3)
29+
if [ $modelname == "Pix2pix" ]; then
30+
sed -i "s/lite_train_lite_infer=10/lite_train_lite_infer=1/g" $FILENAME
31+
fi
32+
33+
if [ $modelname == "edvr" ]; then
34+
sed -i "s/lite_train_lite_infer=100/lite_train_lite_infer=10/g" $FILENAME
35+
fi
36+
2737
# change gpu to npu in execution script
2838
sed -i 's/\"gpu\"/\"npu\"/g' test_tipc/test_train_inference_python.sh
2939
sed -i 's/--gpus/--npus/g' test_tipc/test_train_inference_python.sh

0 commit comments

Comments
 (0)