From 43fce425ac01464b9a63f31898c683c7d9199829 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Fri, 7 Jan 2022 07:48:24 +0000
Subject: [PATCH] fix: change the returned result from dict to list

---
 ppcls/arch/backbone/base/theseus_layer.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/ppcls/arch/backbone/base/theseus_layer.py b/ppcls/arch/backbone/base/theseus_layer.py
index 466b1631..94bc8393 100644
--- a/ppcls/arch/backbone/base/theseus_layer.py
+++ b/ppcls/arch/backbone/base/theseus_layer.py
@@ -112,7 +112,7 @@ class TheseusLayer(nn.Layer):
         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 @@ class TheseusLayer(nn.Layer):
             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,7 +192,7 @@ class TheseusLayer(nn.Layer):
 
         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"):
@@ -200,7 +200,7 @@ class TheseusLayer(nn.Layer):
         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,
-- 
GitLab