From da9fb3d22c5f64e3d4cfb695c41fbb1e92aa1492 Mon Sep 17 00:00:00 2001 From: Binbin Zhang Date: Tue, 7 Dec 2021 10:05:35 +0800 Subject: [PATCH] fix scoring gpu id --- examples/hi_xiaowen/s0/run.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/hi_xiaowen/s0/run.sh b/examples/hi_xiaowen/s0/run.sh index 1542617..3f7967b 100755 --- a/examples/hi_xiaowen/s0/run.sh +++ b/examples/hi_xiaowen/s0/run.sh @@ -98,7 +98,7 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then # Compute posterior score result_dir=$dir/test_$(basename $score_checkpoint) mkdir -p $result_dir - python kws/bin/score.py --gpu 0 \ + python kws/bin/score.py --gpu $gpu_id \ --config $dir/config.yaml \ --test_data data/test/data.list \ --batch_size 256 \