diff --git a/ppcls/arch/backbone/base/theseus_layer.py b/ppcls/arch/backbone/base/theseus_layer.py
index fe0fac15e4457c941f3f8aa3b67858c0e2904e97..8ef7913f89a55586698ffca985dc3eb61ed4ff24 100644
--- a/ppcls/arch/backbone/base/theseus_layer.py
+++ b/ppcls/arch/backbone/base/theseus_layer.py
@@ -15,7 +15,6 @@ 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)
         if return_patterns is not None:
             self._update_res(return_patterns)
 
@@ -48,12 +47,6 @@ class TheseusLayer(nn.Layer):
                     self._sub_layers[layer_i].register_forward_post_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,
                     recursive=True):
         for k in self._sub_layers.keys():