未验证 提交 f9e8668e 编写于 作者: C cuicheng01 提交者: GitHub

fix some architectures bugs (#545)

* Update xception.py

* Update vgg.py

* Update regnet.py
上级 dedd3b0e
...@@ -144,7 +144,7 @@ class BottleneckBlock(nn.Layer): ...@@ -144,7 +144,7 @@ class BottleneckBlock(nn.Layer):
self.se_block = SELayer( self.se_block = SELayer(
num_channels=w_b, num_channels=w_b,
num_filters=w_b, num_filters=w_b,
reduction_channels=w_se, reduction_ratio=w_se,
name=name + "_branch2se") name=name + "_branch2se")
self.conv2 = ConvBNLayer( self.conv2 = ConvBNLayer(
num_channels=w_b, num_channels=w_b,
......
...@@ -80,7 +80,7 @@ class VGGNet(nn.Layer): ...@@ -80,7 +80,7 @@ class VGGNet(nn.Layer):
} }
assert self.layers in self.vgg_configure.keys(), \ assert self.layers in self.vgg_configure.keys(), \
"supported layers are {} but input layer is {}".format( "supported layers are {} but input layer is {}".format(
vgg_configure.keys(), layers) self.vgg_configure.keys(), layers)
self.groups = self.vgg_configure[self.layers] self.groups = self.vgg_configure[self.layers]
self._conv_block_1 = ConvBlock(3, 64, self.groups[0], name="conv1_") self._conv_block_1 = ConvBlock(3, 64, self.groups[0], name="conv1_")
......
...@@ -6,6 +6,7 @@ from paddle.nn import Conv2D, BatchNorm, Linear, Dropout ...@@ -6,6 +6,7 @@ from paddle.nn import Conv2D, BatchNorm, Linear, Dropout
from paddle.nn import AdaptiveAvgPool2D, MaxPool2D, AvgPool2D from paddle.nn import AdaptiveAvgPool2D, MaxPool2D, AvgPool2D
from paddle.nn.initializer import Uniform from paddle.nn.initializer import Uniform
import math import math
import sys
__all__ = ['Xception41', 'Xception65', 'Xception71'] __all__ = ['Xception41', 'Xception65', 'Xception71']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册