提交 b53b990e 编写于 作者: C chenguowei01

update resnet_vd.py

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