提交 e575b54f 编写于 作者: C chenguowei01

update layer_libs.py

上级 caaff468
...@@ -25,13 +25,13 @@ class ConvBNRelu(nn.Layer): ...@@ -25,13 +25,13 @@ class ConvBNRelu(nn.Layer):
super(ConvBNRelu, self).__init__() super(ConvBNRelu, self).__init__()
self.conv = Conv2d(in_channels, out_channels, kernel_size, **kwargs) self._conv = Conv2d(in_channels, out_channels, kernel_size, **kwargs)
self.batch_norm = BatchNorm(out_channels) self._batch_norm = BatchNorm(out_channels)
def forward(self, x): def forward(self, x):
x = self.conv(x) x = self._conv(x)
x = self.batch_norm(x) x = self._batch_norm(x)
x = F.relu(x) x = F.relu(x)
return x return x
...@@ -41,13 +41,13 @@ class ConvBN(nn.Layer): ...@@ -41,13 +41,13 @@ class ConvBN(nn.Layer):
super(ConvBN, self).__init__() super(ConvBN, self).__init__()
self.conv = Conv2d(in_channels, out_channels, kernel_size, **kwargs) self._conv = Conv2d(in_channels, out_channels, kernel_size, **kwargs)
self.batch_norm = BatchNorm(out_channels) self._batch_norm = BatchNorm(out_channels)
def forward(self, x): def forward(self, x):
x = self.conv(x) x = self._conv(x)
x = self.batch_norm(x) x = self._batch_norm(x)
return x return x
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册