未验证 提交 378ed960 编写于 作者: L littletomatodonkey 提交者: GitHub

fix mv2 prefix name (#591)

上级 114b1e3b
...@@ -149,7 +149,7 @@ class InvresiBlocks(nn.Layer): ...@@ -149,7 +149,7 @@ class InvresiBlocks(nn.Layer):
class MobileNet(nn.Layer): class MobileNet(nn.Layer):
def __init__(self, class_dim=1000, scale=1.0): def __init__(self, class_dim=1000, scale=1.0, prefix_name="", **args):
super(MobileNet, self).__init__() super(MobileNet, self).__init__()
self.scale = scale self.scale = scale
self.class_dim = class_dim self.class_dim = class_dim
...@@ -170,7 +170,7 @@ class MobileNet(nn.Layer): ...@@ -170,7 +170,7 @@ class MobileNet(nn.Layer):
filter_size=3, filter_size=3,
stride=2, stride=2,
padding=1, padding=1,
name="conv1_1") name=prefix_name + "conv1_1")
self.block_list = [] self.block_list = []
i = 1 i = 1
...@@ -179,14 +179,14 @@ class MobileNet(nn.Layer): ...@@ -179,14 +179,14 @@ class MobileNet(nn.Layer):
t, c, n, s = layer_setting t, c, n, s = layer_setting
i += 1 i += 1
block = self.add_sublayer( block = self.add_sublayer(
"conv" + str(i), prefix_name + "conv" + str(i),
sublayer=InvresiBlocks( sublayer=InvresiBlocks(
in_c=in_c, in_c=in_c,
t=t, t=t,
c=int(c * scale), c=int(c * scale),
n=n, n=n,
s=s, s=s,
name="conv" + str(i))) name=prefix_name + "conv" + str(i)))
self.block_list.append(block) self.block_list.append(block)
in_c = int(c * scale) in_c = int(c * scale)
...@@ -197,15 +197,15 @@ class MobileNet(nn.Layer): ...@@ -197,15 +197,15 @@ class MobileNet(nn.Layer):
filter_size=1, filter_size=1,
stride=1, stride=1,
padding=0, padding=0,
name="conv9") name=prefix_name + "conv9")
self.pool2d_avg = AdaptiveAvgPool2D(1) self.pool2d_avg = AdaptiveAvgPool2D(1)
self.out = Linear( self.out = Linear(
self.out_c, self.out_c,
class_dim, class_dim,
weight_attr=ParamAttr(name="fc10_weights"), weight_attr=ParamAttr(name=prefix_name + "fc10_weights"),
bias_attr=ParamAttr(name="fc10_offset")) bias_attr=ParamAttr(name=prefix_name + "fc10_offset"))
def forward(self, inputs): def forward(self, inputs):
y = self.conv1(inputs, if_act=True) y = self.conv1(inputs, if_act=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册