fix label smooth bug (#1192)

cf9fc936 · littletomatodonkey · GitHub · d2d89bec · cf9fc936
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/loss/celoss.py ppcls/loss/celoss.py +1 -1

未找到文件。
--- a/ppcls/loss/celoss.py
+++ b/ppcls/loss/celoss.py
@@ -29,7 +29,7 @@ class CELoss(nn.Layer):
        self.epsilon = epsilon

    def _labelsmoothing(self, target, class_num):
-        if target.shape[-1] != class_num:
+        if target.ndim == 1 or target.shape[-1] != class_num:
            one_hot_target = F.one_hot(target, class_num)
        else:
            one_hot_target = target