提交 b53b990e 编写于 作者: C chenguowei01

update resnet_vd.py

上级 3588964d
...@@ -365,7 +365,7 @@ class ResNet_vd(fluid.dygraph.Layer): ...@@ -365,7 +365,7 @@ class ResNet_vd(fluid.dygraph.Layer):
y = self.pool2d_avg(y) y = self.pool2d_avg(y)
y = fluid.layers.reshape(y, shape=[-1, self.pool2d_avg_channels]) y = fluid.layers.reshape(y, shape=[-1, self.pool2d_avg_channels])
y = self.out(y) y = self.out(y)
return y, feat_list return feat_list
# def init_weight(self, pretrained_model=None): # def init_weight(self, pretrained_model=None):
...@@ -387,6 +387,7 @@ class ResNet_vd(fluid.dygraph.Layer): ...@@ -387,6 +387,7 @@ class ResNet_vd(fluid.dygraph.Layer):
pretrained_model)) pretrained_model))
@manager.BACKBONES.add_component
def ResNet18_vd(**args): def ResNet18_vd(**args):
model = ResNet_vd(layers=18, **args) model = ResNet_vd(layers=18, **args)
return model return model
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册