shixian.shi
2024-01-29 c47ad73e46bbaad7b2b003ecd890dc0e8f537172
funasr/losses/label_smoothing_loss.py
@@ -8,7 +8,7 @@
import torch
from torch import nn
from funasr.modules.nets_utils import make_pad_mask
from funasr.models.transformer.utils.nets_utils import make_pad_mask
class LabelSmoothingLoss(nn.Module):