From 79c1b7cd336a11b2d896a4e83334217fb54f2ec9 Mon Sep 17 00:00:00 2001 From: YiwenShaoStephen Date: Wed, 26 Aug 2020 16:49:23 -0400 Subject: [PATCH] minor fix --- examples/wsj/run.sh | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/examples/wsj/run.sh b/examples/wsj/run.sh index 8303619..17afd62 100755 --- a/examples/wsj/run.sh +++ b/examples/wsj/run.sh @@ -157,15 +157,14 @@ fi if [ ${stage} -le 6 ]; then echo "Stage 6: Dumping Posteriors for Test Data" - path=$dir/$checkpoint log_file=$dir/logs/dump_$test_set.log result_file=$test_set/posteriors.ark mkdir -p $dir/$test_set - $cuda_cmd $log_file test.py \ - --test data/test_${type}${unit}.json \ - --exp $dir \ - --model model_best.pth.tar \ - --results $result_file + python3 test.py \ + --test data/test_${type}${unit}.json \ + --model model_best.pth.tar \ + --results $result_file \ + --exp $dir 2>&1 | tee $log_file fi if [ ${stage} -le 7 ]; then