mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update attention_decoder.py (#1681)

Browse Source
This commit is contained in:
zr_jin 2024-07-06 09:01:34 +08:00 committed by GitHub
parent 325a825841
commit 2d64228efa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/zipformer/attention_decoder.py
View File

@ -22,11 +22,11 @@ from typing import List, Optional
import k2 import k2
import torch import torch
import torch.nn as nn import torch.nn as nn
from label_smoothing import LabelSmoothingLoss from label_smoothing import LabelSmoothingLoss
from icefall.utils import add_eos, add_sos, make_pad_mask
from scaling import penalize_abs_values_gt from scaling import penalize_abs_values_gt
from icefall.utils import add_eos, add_sos, make_pad_mask
class AttentionDecoderModel(nn.Module): class AttentionDecoderModel(nn.Module):
""" """