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

Update mobilenet_v1.py

上级 f5ab94b4
......@@ -141,7 +141,7 @@ class MobileNet(TheseusLayer):
self.avg_pool = AdaptiveAvgPool2D(1)
self.flatten = Flatten(start_axis=1, stop_axis=-1)
self.out = Linear(
self.fc = Linear(
int(1024 * scale),
class_num,
weight_attr=ParamAttr(initializer=KaimingNormal()))
......@@ -151,7 +151,7 @@ class MobileNet(TheseusLayer):
x = self.blocks(x)
x = self.avg_pool(x)
x = self.flatten(x)
x = self.out(x)
x = self.fc(x)
return x
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册