提交 10c5773b 编写于 作者: L LDOUBLEV

fix

上级 62a01f7e
...@@ -120,6 +120,7 @@ class CALayer(nn.Layer): ...@@ -120,6 +120,7 @@ class CALayer(nn.Layer):
def __init__(self, in_channels, out_channels, kernel_size, shortcut=True): def __init__(self, in_channels, out_channels, kernel_size, shortcut=True):
super(CALayer, self).__init__() super(CALayer, self).__init__()
weight_attr = paddle.nn.initializer.KaimingUniform() weight_attr = paddle.nn.initializer.KaimingUniform()
self.out_channels = out_channels
self.in_conv = nn.Conv2D( self.in_conv = nn.Conv2D(
in_channels=in_channels, in_channels=in_channels,
out_channels=self.out_channels, out_channels=self.out_channels,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册