Fix sample code of poisson_nll_loss (#53551)

PaddlePaddle · May 8, 2023 · 3fd2e76 · 3fd2e76
1 parent acefdeb
commit 3fd2e76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py
@@ -1541,7 +1541,7 @@ def poisson_nll_loss(
 
             input = paddle.randn([5, 2], dtype=paddle.float32)
             label = paddle.randn([5, 2], dtype=paddle.float32)
-            loss = F.poisson_nll_loss(input, label, log_input=True, reduction='None')
+            loss = F.poisson_nll_loss(input, label, log_input=True, reduction='none')
             print(loss)
             loss = F.poisson_nll_loss(input, label, reduction='mean')
             print(loss)