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

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

@ -94,7 +94,7 @@ class WrapLayer(TheseusLayer):
return self.name return self.name
def forward(self, *inputs, **kwargs): def forward(self, *inputs, **kwargs):
self.sub_layer(*inputs, **kwargs) return self.sub_layer(*inputs, **kwargs)
def update_res(self, return_patterns): def update_res(self, return_patterns):
if not return_patterns or not isinstance(self.sub_layer, (nn.Sequential, nn.LayerList)): if not return_patterns or not isinstance(self.sub_layer, (nn.Sequential, nn.LayerList)):