From 2b150a2ffa5ea3b39ab8b8494154b24d545133ea Mon Sep 17 00:00:00 2001 From: weishengyu Date: Tue, 25 May 2021 16:51:04 +0800 Subject: [PATCH] dbg --- ppcls/arch/backbone/base/theseus_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/arch/backbone/base/theseus_layer.py b/ppcls/arch/backbone/base/theseus_layer.py index 875131cb..58965f83 100644 --- a/ppcls/arch/backbone/base/theseus_layer.py +++ b/ppcls/arch/backbone/base/theseus_layer.py @@ -15,7 +15,7 @@ class TheseusLayer(nn.Layer): def __init__(self, *args, return_patterns=None, **kwargs): super(TheseusLayer, self).__init__() self.res_dict = None - self.register_forward_post_hook(self._disconnect_res_dict_hook) + # self.register_forward_post_hook(self._disconnect_res_dict_hook) if return_patterns is not None: self._update_res(return_patterns) -- GitLab