Skip to content

Commit

Permalink
Update attention_decoder.py
Browse files Browse the repository at this point in the history
  • Loading branch information
JinZr committed Jul 5, 2024
1 parent f76afff commit 3cad98d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions egs/librispeech/ASR/zipformer/attention_decoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
"""
Expand Down

0 comments on commit 3cad98d

Please sign in to comment.