We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a99e506 commit 029edf8Copy full SHA for 029edf8
test_tipc/test_train_inference_python_npu.sh
@@ -24,6 +24,16 @@ sed -i "s/--device:gpu/--device:npu/g" $FILENAME
24
sed -i "s/--benchmark:True/--benchmark:False/g" $FILENAME
25
dataline=`cat $FILENAME`
26
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
36
37
# change gpu to npu in execution script
38
sed -i 's/\"gpu\"/\"npu\"/g' test_tipc/test_train_inference_python.sh
39
sed -i 's/--gpus/--npus/g' test_tipc/test_train_inference_python.sh
0 commit comments