modifying some implmentations of mdtc to get the model torch-scripting through

2021-11-26 23:54:32 +08:00 · 2021-11-26 23:54:32 +08:00 · b5dde01df3
commit b5dde01df3
parent 5bc7c8d64e
1 changed files with 2 additions and 1 deletions
--- a/examples/hi_xiaowen/s0/run.sh
+++ b/examples/hi_xiaowen/s0/run.sh
@ -102,7 +102,8 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
    --test_data data/test/data.list \
    --batch_size 256 \
    --checkpoint $score_checkpoint \
-    --score_file $result_dir/score.txt
+    --score_file $result_dir/score.txt \
+    --num_workers 8
 fi

 if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then