mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

dbg

Browse Source
This commit is contained in:
weishengyu 2021-08-08 15:56:57 +08:00
parent 2cb3bf66dd
commit 88a72b6735
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/arch/backbone/base/theseus_layer.py
View File

@ -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):