change the comments of yolov3_loss to yolo_loss

PaddlePaddle · Dec 6, 2022 · 1d116c5 · 1d116c5
1 parent 36fffe9
commit 1d116c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_device_guard.py b/python/paddle/fluid/tests/unittests/test_device_guard.py
@@ -127,7 +127,7 @@ def test_cpu_only_op(self):
             ]
             anchor_mask = [0, 1, 2]
             with paddle.static.device_guard("gpu"):
-                # yolov3_loss only has cpu kernel, so its cpu kernel will be executed
+                # yolo_loss only has cpu kernel, so its cpu kernel will be executed
                 loss = paddle.vision.ops.yolo_loss(
                     x=x,
                     gt_box=gt_box,

diff --git a/python/paddle/fluid/tests/unittests/xpu/test_device_guard_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_device_guard_xpu.py
@@ -133,7 +133,7 @@ def test_cpu_only_op(self):
             ]
             anchor_mask = [0, 1, 2]
             with paddle.static.device_guard("xpu"):
-                # yolov3_loss only has cpu kernel, so its cpu kernel will be executed
+                # yolo_loss has cpu kernel, so its cpu kernel will be executed
                 loss = paddle.vision.ops.yolo_loss(
                     x=x,
                     gt_box=gt_box,