未验证 提交 5ec82d89 编写于 作者: W wangguanzhong 提交者: GitHub

update Conv2D (#1594)

上级 3ab53043
......@@ -20,7 +20,7 @@ class ConvBNLayer(nn.Layer):
name=None):
super(ConvBNLayer, self).__init__()
self.conv = nn.Conv2d(
self.conv = nn.Conv2D(
in_channels=ch_in,
out_channels=ch_out,
kernel_size=filter_size,
......
......@@ -128,7 +128,7 @@ class YOLOv3Head(nn.Layer):
name = 'yolo_output.{}'.format(i)
yolo_out = self.add_sublayer(
name,
nn.Conv2d(
nn.Conv2D(
in_channels=1024 // (2**i),
out_channels=num_filters,
kernel_size=1,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册