From b53b990e8993bd54fba3853d74da06518533e40d Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Mon, 14 Sep 2020 16:20:31 +0800 Subject: [PATCH] update resnet_vd.py --- dygraph/models/architectures/resnet_vd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dygraph/models/architectures/resnet_vd.py b/dygraph/models/architectures/resnet_vd.py index 58293450..2d117cb2 100644 --- a/dygraph/models/architectures/resnet_vd.py +++ b/dygraph/models/architectures/resnet_vd.py @@ -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 -- GitLab