zhaomingwork
2023-05-15 00de4511009a7fbbb88142b2b2a0faeaa0e3e7d9
funasr/losses/label_smoothing_loss.py
@@ -97,7 +97,7 @@
        normalize_length=False,
        criterion=nn.NLLLoss(reduction='none'),
    ):
        """Construct an LabelSmoothingLoss object."""
        """Construct an NllLoss object."""
        super(NllLoss, self).__init__()
        self.criterion = criterion
        self.padding_idx = padding_idx