From d92b6781b9e6b719b9d12cab05ad9880a40d4d74 Mon Sep 17 00:00:00 2001 From: Yifan Yang Date: Thu, 15 Jun 2023 17:58:23 +0800 Subject: [PATCH] update --- egs/librispeech/ASR/zipformer/decoder.py | 1 - egs/librispeech/ASR/zipformer/model.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/decoder.py b/egs/librispeech/ASR/zipformer/decoder.py index 45432d570..e8db988f6 100644 --- a/egs/librispeech/ASR/zipformer/decoder.py +++ b/egs/librispeech/ASR/zipformer/decoder.py @@ -58,7 +58,6 @@ class Decoder(nn.Module): self.embedding = nn.Embedding( num_embeddings=vocab_size, embedding_dim=decoder_dim, - padding_idx=blank_id, ) # the balancers are to avoid any drift in the magnitude of the # embeddings, which would interact badly with parameter averaging. diff --git a/egs/librispeech/ASR/zipformer/model.py b/egs/librispeech/ASR/zipformer/model.py index 9b7494972..0c3ea6a86 100644 --- a/egs/librispeech/ASR/zipformer/model.py +++ b/egs/librispeech/ASR/zipformer/model.py @@ -333,7 +333,7 @@ class AsrModel(nn.Module): simple_loss, pruned_loss = self.forward_transducer( encoder_out=encoder_out, encoder_out_lens=encoder_out_lens, - y=y, + y=y.to(x.device), y_lens=y_lens, prune_range=prune_range, am_scale=am_scale,