remove disconnect
parent
3b62e4d401
commit
56c513fdc7
|
@ -15,7 +15,6 @@ class TheseusLayer(nn.Layer):
|
|||
def __init__(self, *args, **kwargs):
|
||||
super(TheseusLayer, self).__init__()
|
||||
self.res_dict = None
|
||||
self.register_forward_post_hook(self._disconnect_res_dict_hook)
|
||||
|
||||
def forward(self, *input, res_dict=None, **kwargs):
|
||||
if res_dict is not None:
|
||||
|
@ -52,9 +51,6 @@ class TheseusLayer(nn.Layer):
|
|||
if self.res_dict is not None:
|
||||
self.res_dict[layer.full_name()] = output
|
||||
|
||||
def _disconnect_res_dict_hook(self, *args, **kwargs):
|
||||
self.res_dict = None
|
||||
|
||||
def replace_sub(self, layer_name_pattern, replace_function, recursive=True):
|
||||
for layer_i in self._sub_layers:
|
||||
layer_name = self._sub_layers[layer_i].full_name()
|
||||
|
|
Loading…
Reference in New Issue