fix: change the returned result from dict to list

Author: TingquanGao

ppcls/arch/backbone/base/theseus_layer.py

@@ -112,7 +112,7 @@ def rep_func(layer: nn.Layer, pattern: str):
         if not isinstance(layer_name_pattern, list):
             layer_name_pattern = [layer_name_pattern]
 
-        handle_res_dict = {}
+        hit_layer_pattern_list = []
         for pattern in layer_name_pattern:
             # parse pattern to find target layer and its parent
             layer_list = parse_pattern_str(pattern=pattern, parent_layer=self)
@@ -133,8 +133,8 @@ def rep_func(layer: nn.Layer, pattern: str):
             else:
                 setattr(sub_layer_parent, sub_layer_name, new_sub_layer)
 
-            handle_res_dict[pattern] = new_sub_layer
-        return handle_res_dict
+            hit_layer_pattern_list.append(pattern)
+        return hit_layer_pattern_list
 
     def stop_after(self, stop_layer_name: str) -> bool:
         """stop forward and backward after 'stop_layer_name'.
@@ -192,15 +192,15 @@ def __call__(self, layer, pattern):
 
         handle_func = Handler(self.res_dict)
 
-        res_dict = self.upgrade_sublayer(
+        hit_layer_pattern_list = self.upgrade_sublayer(
             return_patterns, handle_func=handle_func)
 
         if hasattr(self, "hook_remove_helper"):
             self.hook_remove_helper.remove()
         self.hook_remove_helper = self.register_forward_post_hook(
             self._return_dict_hook)
 
-        return res_dict
+        return hit_layer_pattern_list
 
 
 def set_identity(parent_layer: nn.Layer,