modifying some implmentations of mdtc to get the model torch-scripting through
This commit is contained in:
parent
5bc7c8d64e
commit
b5dde01df3
@ -102,7 +102,8 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
|
|||||||
--test_data data/test/data.list \
|
--test_data data/test/data.list \
|
||||||
--batch_size 256 \
|
--batch_size 256 \
|
||||||
--checkpoint $score_checkpoint \
|
--checkpoint $score_checkpoint \
|
||||||
--score_file $result_dir/score.txt
|
--score_file $result_dir/score.txt \
|
||||||
|
--num_workers 8
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
|
if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user