mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #868 from weisy11/develop 

remove useless comments
pull/869/head
cuicheng01 2021-06-16 22:18:48 +08:00 committed by GitHub
parent 96d64c41e2 e11cc5c8c1
commit 1691aa5a7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,7 +15,6 @@ class TheseusLayer(nn.Layer):
def __init__(self, *args, return_patterns=None, **kwargs): def __init__(self, *args, return_patterns=None, **kwargs):
super(TheseusLayer, self).__init__() super(TheseusLayer, self).__init__()
self.res_dict = None self.res_dict = None
# self.register_forward_post_hook(self._disconnect_res_dict_hook)
if return_patterns is not None: if return_patterns is not None:
self._update_res(return_patterns) self._update_res(return_patterns)
@ -48,12 +47,6 @@ class TheseusLayer(nn.Layer):
self._sub_layers[layer_i].register_forward_post_hook( self._sub_layers[layer_i].register_forward_post_hook(
self._save_sub_res_hook) self._save_sub_res_hook)
# def _save_sub_res_hook(self, layer, input, output):
# self.res_dict[layer.full_name()] = output
#
# def _disconnect_res_dict_hook(self, input, output):
# self.res_dict = None
def replace_sub(self, layer_name_pattern, replace_function, def replace_sub(self, layer_name_pattern, replace_function,
recursive=True): recursive=True):
for k in self._sub_layers.keys(): for k in self._sub_layers.keys():