From 9aa6bfc7e1cfce657109789995d153b6bcdf74d7 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Tue, 8 Mar 2022 17:42:29 +0800 Subject: [PATCH] fix yolov3 return value in dygraph mode. test=develop (#40185) --- python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py | 1 + python/paddle/vision/ops.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py b/python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py index 1ec1d1527e..3f0e4f7a40 100644 --- a/python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py +++ b/python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py @@ -305,6 +305,7 @@ class TestYolov3LossDygraph(unittest.TestCase): use_label_smooth=True, scale_x_y=1.) assert loss is not None + assert loss.shape == [2] paddle.enable_static() diff --git a/python/paddle/vision/ops.py b/python/paddle/vision/ops.py index 03060e92bd..4983ca49ac 100644 --- a/python/paddle/vision/ops.py +++ b/python/paddle/vision/ops.py @@ -195,7 +195,7 @@ def yolo_loss(x, """ if in_dygraph_mode() and gt_score is None: - loss = _C_ops.yolov3_loss( + loss, _, _ = _C_ops.yolov3_loss( x, gt_box, gt_label, 'anchors', anchors, 'anchor_mask', anchor_mask, 'class_num', class_num, 'ignore_thresh', ignore_thresh, 'downsample_ratio', downsample_ratio, 'use_label_smooth', -- GitLab