未验证 提交 f5ab94b4 编写于 作者: B Bin Lu 提交者: GitHub

Update mobilenet_v1.py

上级 5d155c8f
......@@ -58,15 +58,15 @@ class ConvBNLayer(TheseusLayer):
initializer=KaimingNormal()),
bias_attr=False)
self._batch_norm = BatchNorm(
self._bn = BatchNorm(
num_filters)
self._activation = ReLU()
self._relu = ReLU()
def forward(self, x):
x = self._conv(x)
x = self._batch_norm(x)
x = self._activation(x)
x = self._bn(x)
x = self._relu(x)
return x
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册