From 2d64228efae6ebb85b68a956942374a4801548ae Mon Sep 17 00:00:00 2001 From: zr_jin Date: Sat, 6 Jul 2024 09:01:34 +0800 Subject: [PATCH] Update attention_decoder.py (#1681) --- egs/librispeech/ASR/zipformer/attention_decoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/attention_decoder.py b/egs/librispeech/ASR/zipformer/attention_decoder.py index 71be2d1ebb..81682e87b5 100644 --- a/egs/librispeech/ASR/zipformer/attention_decoder.py +++ b/egs/librispeech/ASR/zipformer/attention_decoder.py @@ -22,11 +22,11 @@ import k2 import torch import torch.nn as nn - from label_smoothing import LabelSmoothingLoss -from icefall.utils import add_eos, add_sos, make_pad_mask from scaling import penalize_abs_values_gt +from icefall.utils import add_eos, add_sos, make_pad_mask + class AttentionDecoderModel(nn.Module): """