Update loss.py

fa4805b4 · HydrogenSulfate · chajchaj · 39e81532 · fa4805b4
显示空白变更内容
内联并排

Showing with 4 addition and 3 deletion

python/paddle/nn/functional/loss.py python/paddle/nn/functional/loss.py +4 -3

未找到文件。
--- a/python/paddle/nn/functional/loss.py
+++ b/python/paddle/nn/functional/loss.py
@@ -1815,7 +1815,8 @@ def cross_entropy(input,
 		            [1], dtype=label.dtype),
 	    	    label)
            if (paddle.numel(paddle.nonzero(valid_label < 0)) > 0) or (
-                    paddle.numel(paddle.nonzero(valid_label >= input.shape[-1])) > 0):
+                    paddle.numel(
+						paddle.nonzero(valid_label >= input.shape[-1])) > 0):
                invalid_label = paddle.gather_nd(
                    input, paddle.nonzero(valid_label < 0))
                if paddle.numel(invalid_label) > 0: