From e2ad07f04f1f48abd248a9808db3bb8ffd7562e0 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Mon, 15 Apr 2019 06:31:15 +0000
Subject: [PATCH] refine yolov3_loss param name.

---
 PaddleCV/yolov3/models/yolov3.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/PaddleCV/yolov3/models/yolov3.py b/PaddleCV/yolov3/models/yolov3.py
index 55c0667b..87396833 100644
--- a/PaddleCV/yolov3/models/yolov3.py
+++ b/PaddleCV/yolov3/models/yolov3.py
@@ -139,9 +139,9 @@ class YOLOv3(object):
             if self.is_train:
                 loss = fluid.layers.yolov3_loss(
                         x=out,
-                        gtbox=self.gtbox,
-                        gtlabel=self.gtlabel,
-                        gtscore=self.gtscore,
+                        gt_box=self.gtbox,
+                        gt_label=self.gtlabel,
+                        gt_score=self.gtscore,
                         anchors=cfg.anchors,
                         anchor_mask=anchor_mask,
                         class_num=cfg.class_num,
-- 
GitLab