From 4ffec0e7d2ab26316405396648755f136b3ff70a Mon Sep 17 00:00:00 2001
From: huangyuxin <huangyuxin@baidu.com>
Date: Thu, 22 Jul 2021 02:59:45 +0000
Subject: [PATCH] add autolog

---
 examples/aishell/s0/run.sh | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/examples/aishell/s0/run.sh b/examples/aishell/s0/run.sh
index c9708dcc9..6f4c23bd4 100755
--- a/examples/aishell/s0/run.sh
+++ b/examples/aishell/s0/run.sh
@@ -2,9 +2,9 @@
 set -e
 source path.sh
 
-gpus=0,1,2,3
-stage=0
-stop_stage=100
+gpus=2
+stage=3
+stop_stage=3
 conf_path=conf/deepspeech2.yaml
 avg_num=1
 
@@ -31,10 +31,10 @@ fi
 
 if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
     # test ckpt avg_n
-    CUDA_VISIBLE_DEVICES=0 ./local/test.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} || exit -1
+    CUDA_VISIBLE_DEVICES=${gpus} ./local/test.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} || exit -1
 fi
 
 if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
     # export ckpt avg_n
-    CUDA_VISIBLE_DEVICES=0 ./local/export.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} exp/${ckpt}/checkpoints/${avg_ckpt}.jit
+    CUDA_VISIBLE_DEVICES=${gpus} ./local/export.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} exp/${ckpt}/checkpoints/${avg_ckpt}.jit
 fi