From f5d447ca23b87541e2f78e2012a244cb9d8a0bbf Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Fri, 4 Mar 2022 14:12:20 +0000 Subject: [PATCH] fix yolov3 return value in dygraph mode. test=develop --- 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 1ec1d1527e178a..3f0e4f7a4002a2 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 @@ def test_dygraph(self): 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 03060e92bdb69b..4983ca49ac32fb 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',