Skip to content

Commit cdb617d

Browse files
authored
Fix npu tipc to update snapshot_config (#782)
* update tipc scripts for time-consuming model on npu
1 parent 029edf8 commit cdb617d

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

test_tipc/test_train_inference_python_npu.sh

+8-5
Original file line numberDiff line numberDiff line change
@@ -24,24 +24,27 @@ 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+
# parser params
28+
IFS=$'\n'
29+
lines=(${dataline})
30+
2731
# change total iters/epochs for npu/xpu to accelaration
28-
modelname=$(echo $FILENAME | cut -d '/' -f3)
32+
modelname=$(func_parser_value "${lines[1]}")
33+
echo $modelname
2934
if [ $modelname == "Pix2pix" ]; then
3035
sed -i "s/lite_train_lite_infer=10/lite_train_lite_infer=1/g" $FILENAME
36+
sed -i "s/-o log_config.interval=1/-o log_config.interval=1 snapshot_config.interval=1/g" $FILENAME
3137
fi
3238

3339
if [ $modelname == "edvr" ]; then
3440
sed -i "s/lite_train_lite_infer=100/lite_train_lite_infer=10/g" $FILENAME
41+
sed -i "s/snapshot_config.interval=25/snapshot_config.interval=5/g" $FILENAME
3542
fi
3643

3744
# change gpu to npu in execution script
3845
sed -i 's/\"gpu\"/\"npu\"/g' test_tipc/test_train_inference_python.sh
3946
sed -i 's/--gpus/--npus/g' test_tipc/test_train_inference_python.sh
4047

41-
# parser params
42-
IFS=$'\n'
43-
lines=(${dataline})
44-
4548
# pass parameters to test_train_inference_python.sh
4649
cmd="bash test_tipc/test_train_inference_python.sh ${FILENAME} $2"
4750
echo $cmd

0 commit comments

Comments
 (0)