From 6c5fd6f6480cfab047a3af0c2f8c078293f92e7f Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Mon, 9 May 2022 16:41:46 +0800 Subject: [PATCH] minor fix of train.py --- egs/librispeech/ASR/conv_emformer_transducer/train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/egs/librispeech/ASR/conv_emformer_transducer/train.py b/egs/librispeech/ASR/conv_emformer_transducer/train.py index 8a0eecc6b..5152be1a1 100755 --- a/egs/librispeech/ASR/conv_emformer_transducer/train.py +++ b/egs/librispeech/ASR/conv_emformer_transducer/train.py @@ -394,6 +394,7 @@ def get_decoder_model(params: AttributeDict) -> nn.Module: vocab_size=params.vocab_size, embedding_dim=params.embedding_dim, blank_id=params.blank_id, + unk_id=params.unk_id, context_size=params.context_size, ) return decoder @@ -829,6 +830,7 @@ def run(rank, world_size, args): # is defined in local/train_bpe_model.py params.blank_id = sp.piece_to_id("") + params.unk_id = sp.piece_to_id("") params.vocab_size = sp.get_piece_size() logging.info(params)