diff --git a/ppdet/modeling/losses/yolo_loss.py b/ppdet/modeling/losses/yolo_loss.py index 6823c024bc5aea54c4f9b5195da03cb16dfee8a3..197276c4474495656fa14d204d47ef0467f5cbd2 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 = []