fixed registry for @register_loss(LabelSmoothingCrossEntropyLoss) (#1475

)
Deci-AI · Sep 20, 2023 · 941817b · 941817b
1 parent 1b558ed
commit 941817b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/super_gradients/training/losses/label_smoothing_cross_entropy_loss.py b/src/super_gradients/training/losses/label_smoothing_cross_entropy_loss.py
@@ -113,5 +113,6 @@ def forward(self, input, target, smooth_dist=None):
 
 
 @deprecated(deprecated_since="3.2.1", removed_from="3.5.0", target=CrossEntropyLoss)
+@register_loss("LabelSmoothingCrossEntropyLoss")
 class LabelSmoothingCrossEntropyLoss(CrossEntropyLoss):
     ...