From 88a72b673517363651bb3dda57423458b51ef0f1 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Sun, 8 Aug 2021 15:56:57 +0800 Subject: [PATCH] dbg --- ppcls/arch/backbone/base/theseus_layer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/arch/backbone/base/theseus_layer.py b/ppcls/arch/backbone/base/theseus_layer.py index ae0075bb..41599cd9 100644 --- a/ppcls/arch/backbone/base/theseus_layer.py +++ b/ppcls/arch/backbone/base/theseus_layer.py @@ -94,7 +94,7 @@ class WrapLayer(TheseusLayer): return self.name def forward(self, *inputs, **kwargs): - self.sub_layer(*inputs, **kwargs) + return self.sub_layer(*inputs, **kwargs) def update_res(self, return_patterns): if not return_patterns or not isinstance(self.sub_layer, (nn.Sequential, nn.LayerList)): @@ -104,7 +104,7 @@ class WrapLayer(TheseusLayer): self.sub_layer._sub_layers[layer_i] = wrap_theseus(self.sub_layer._sub_layers[layer_i]) self.sub_layer._sub_layers[layer_i].res_dict = self.res_dict self.sub_layer._sub_layers[layer_i].update_res(return_patterns) - + layer_name = self.sub_layer._sub_layers[layer_i].full_name() for return_pattern in return_patterns: if re.match(return_pattern, layer_name): -- GitLab