未验证 提交 9075408a 编写于 作者: G Guanghua Yu 提交者: GitHub

Revert "fix yolov3 last conv quantization (#6813)" (#6879)

This reverts commit 64687d46.
上级 7c41cb35
......@@ -78,7 +78,7 @@ class YOLOv3Head(nn.Layer):
self.num_outputs = len(self.anchors)
self.data_format = data_format
self.yolo_outputs = nn.LayerList()
self.yolo_outputs = []
for i in range(len(self.anchors)):
if self.iou_aware:
......@@ -94,7 +94,9 @@ class YOLOv3Head(nn.Layer):
padding=0,
data_format=data_format,
bias_attr=ParamAttr(regularizer=L2Decay(0.)))
self.yolo_outputs.append(conv)
conv.skip_quant = True
yolo_output = self.add_sublayer(name, conv)
self.yolo_outputs.append(yolo_output)
def parse_anchor(self, anchors, anchor_masks):
self.anchors = [[anchors[i] for i in mask] for mask in anchor_masks]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册