未验证 提交 ccb40836 编写于 作者: H Hongyu Liu 提交者: GitHub

remove conv2d num_channel parameter; test=develop (#2378)

上级 02db4c3b
...@@ -59,7 +59,6 @@ class SimpleImgConvPool(fluid.dygraph.Layer): ...@@ -59,7 +59,6 @@ class SimpleImgConvPool(fluid.dygraph.Layer):
self._conv2d = Conv2D( self._conv2d = Conv2D(
self.full_name(), self.full_name(),
num_channels=num_channels,
num_filters=num_filters, num_filters=num_filters,
filter_size=filter_size, filter_size=filter_size,
stride=conv_stride, stride=conv_stride,
......
...@@ -107,7 +107,6 @@ class ConvBNPool(fluid.dygraph.Layer): ...@@ -107,7 +107,6 @@ class ConvBNPool(fluid.dygraph.Layer):
self.conv_0_layer = Conv2D( self.conv_0_layer = Conv2D(
self.full_name(), self.full_name(),
channels[0],
out_ch[0], out_ch[0],
3, 3,
padding=1, padding=1,
...@@ -119,7 +118,6 @@ class ConvBNPool(fluid.dygraph.Layer): ...@@ -119,7 +118,6 @@ class ConvBNPool(fluid.dygraph.Layer):
self.full_name(), out_ch[0], act=act, is_test=is_test) self.full_name(), out_ch[0], act=act, is_test=is_test)
self.conv_1_layer = Conv2D( self.conv_1_layer = Conv2D(
self.full_name(), self.full_name(),
num_channels=channels[1],
num_filters=out_ch[1], num_filters=out_ch[1],
filter_size=3, filter_size=3,
padding=1, padding=1,
......
...@@ -81,7 +81,6 @@ class ConvBNLayer(fluid.dygraph.Layer): ...@@ -81,7 +81,6 @@ class ConvBNLayer(fluid.dygraph.Layer):
self._conv = Conv2D( self._conv = Conv2D(
self.full_name(), self.full_name(),
num_channels=num_channels,
num_filters=num_filters, num_filters=num_filters,
filter_size=filter_size, filter_size=filter_size,
stride=stride, stride=stride,
......
...@@ -19,7 +19,6 @@ from paddle.fluid.dygraph.base import to_variable ...@@ -19,7 +19,6 @@ from paddle.fluid.dygraph.base import to_variable
class SimpleConvPool(fluid.dygraph.Layer): class SimpleConvPool(fluid.dygraph.Layer):
def __init__(self, def __init__(self,
name_scope, name_scope,
num_channels,
num_filters, num_filters,
filter_size, filter_size,
use_cudnn=False, use_cudnn=False,
...@@ -28,7 +27,6 @@ class SimpleConvPool(fluid.dygraph.Layer): ...@@ -28,7 +27,6 @@ class SimpleConvPool(fluid.dygraph.Layer):
self.batch_size = batch_size self.batch_size = batch_size
self._conv2d = Conv2D( self._conv2d = Conv2D(
self.full_name(), self.full_name(),
num_channels=num_channels,
num_filters=num_filters, num_filters=num_filters,
filter_size=filter_size, filter_size=filter_size,
padding=[1, 1], padding=[1, 1],
...@@ -61,7 +59,6 @@ class CNN(fluid.dygraph.Layer): ...@@ -61,7 +59,6 @@ class CNN(fluid.dygraph.Layer):
self._simple_conv_pool_1 = SimpleConvPool( self._simple_conv_pool_1 = SimpleConvPool(
self.full_name(), self.full_name(),
1,
self.hid_dim, self.hid_dim,
self.win_size, self.win_size,
batch_size=self.batch_size) batch_size=self.batch_size)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册