From 93a6e2eefee7cc0c2fcfc175cfaa2526536d942c Mon Sep 17 00:00:00 2001 From: lhl1001 <87488929+lhl1001@users.noreply.github.com> Date: Fri, 6 Aug 2021 16:13:49 +0800 Subject: [PATCH] Update label_smoothing_loss.py Modify comments --- wenet/transformer/label_smoothing_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wenet/transformer/label_smoothing_loss.py b/wenet/transformer/label_smoothing_loss.py index 4f98fd4f1..36a74e89c 100644 --- a/wenet/transformer/label_smoothing_loss.py +++ b/wenet/transformer/label_smoothing_loss.py @@ -17,7 +17,7 @@ class LabelSmoothingLoss(nn.Module): [ [1.0, 0.0, 0.0], [0.0, 1.0, 0.0], - [1.0, 0.0, 1.0], + [0.0, 0.0, 1.0], ] In the smoothing version CE Loss,some probabilities