Update train-rnn-lm.sh (#1337)

This commit is contained in:
zr_jin 2023-10-25 12:50:35 +08:00 committed by GitHub
parent 1814bbb0e7
commit dcbc7a63e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -37,10 +37,8 @@ if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then
--world-size $world_size \
--use-fp16 0 \
--vocab-size 500 \
\
--lm-data ./data/lm_training_bpe_500/sorted_lm_data.pt \
--lm-data-valid ./data/lm_training_bpe_500/sorted_lm_data-valid.pt \
\
--embedding-dim 800 \
--hidden-dim 200 \
--num-layers 2 \
@ -56,9 +54,7 @@ if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then
--epoch $use_epoch \
--avg $use_avg \
--vocab-size 500 \
\
--lm-data ./data/lm_training_bpe_500/sorted_lm_data-test.pt \
\
--embedding-dim 800 \
--hidden-dim 200 \
--num-layers 2 \