提交 6edc02ef 编写于 作者: D dengkaipeng

temp fix slice op no kernel error

上级 31bca148
...@@ -127,6 +127,7 @@ class YOLOv3Loss(object): ...@@ -127,6 +127,7 @@ class YOLOv3Loss(object):
assert len(outputs) == len(targets), \ assert len(outputs) == len(targets), \
"YOLOv3 output layer number not equal target number" "YOLOv3 output layer number not equal target number"
targets = [target * 1.0 for target in targets]
loss_xys, loss_whs, loss_objs, loss_clss = [], [], [], [] loss_xys, loss_whs, loss_objs, loss_clss = [], [], [], []
if self._iou_loss is not None: if self._iou_loss is not None:
loss_ious = [] loss_ious = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册