Skip to content

Commit 26e6585

Browse files
authored
Merge branch 'master' into master
2 parents 4e8955f + 941817b commit 26e6585

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/super_gradients/training/losses/label_smoothing_cross_entropy_loss.py

+1
Original file line numberDiff line numberDiff line change
@@ -113,5 +113,6 @@ def forward(self, input, target, smooth_dist=None):
113113

114114

115115
@deprecated(deprecated_since="3.2.1", removed_from="3.5.0", target=CrossEntropyLoss)
116+
@register_loss("LabelSmoothingCrossEntropyLoss")
116117
class LabelSmoothingCrossEntropyLoss(CrossEntropyLoss):
117118
...

0 commit comments

Comments
 (0)