From b5dde01df356307adf838fb70779f6de025068c3 Mon Sep 17 00:00:00 2001 From: lxiao336 Date: Fri, 26 Nov 2021 23:54:32 +0800 Subject: [PATCH] modifying some implmentations of mdtc to get the model torch-scripting through --- examples/hi_xiaowen/s0/run.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/hi_xiaowen/s0/run.sh b/examples/hi_xiaowen/s0/run.sh index fe4b782..7c97999 100755 --- 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