fix scoring gpu id

2021-12-07 10:05:35 +08:00 · 2021-12-07 10:05:35 +08:00 · da9fb3d22c
commit da9fb3d22c
parent a718ed39c5
1 changed files with 1 additions and 1 deletions
--- 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 \