From 6edc02ef286a9d138d60b36f8d9959399a7794c8 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Sat, 8 Aug 2020 14:35:59 +0000
Subject: [PATCH] temp fix slice op no kernel error

---
 ppdet/modeling/losses/yolo_loss.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ppdet/modeling/losses/yolo_loss.py b/ppdet/modeling/losses/yolo_loss.py
index 6823c024b..197276c44 100644
--- a/ppdet/modeling/losses/yolo_loss.py
+++ b/ppdet/modeling/losses/yolo_loss.py
@@ -127,6 +127,7 @@ class YOLOv3Loss(object):
         assert len(outputs) == len(targets), \
             "YOLOv3 output layer number not equal target number"
 
+        targets = [target * 1.0 for target in targets]
         loss_xys, loss_whs, loss_objs, loss_clss = [], [], [], []
         if self._iou_loss is not None:
             loss_ious = []
-- 
GitLab