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

fix some architectures bugs (#546)

* Update xception.py

* Update vgg.py

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