dbg

ppcls/arch/backbone/base/theseus_layer.py

@@ -41,7 +41,7 @@ class TheseusLayer(nn.Layer):
             layer_name = self._sub_layers[layer_i].full_name()
             for return_pattern in return_patterns:
                 if re.match(return_pattern, layer_name):
-                    if not isinstance(self._sub_layers[layer_i], TheseusLayer):
+                    if not isinstance(self._sub_layers[layer_i], TheseusLayer) and not isinstance(self, WrapLayer):
                         self._sub_layers[layer_i] = wrap_theseus(self._sub_layers[layer_i], return_patterns)
                     self._sub_layers[layer_i].register_forward_post_hook(
                         self._sub_layers[layer_i]._save_sub_res_hook)