fix bug in tracer

open-mmlab · Sep 30, 2022 · 68d8040 · 68d8040
1 parent 1bcc4da
commit 68d8040
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmrazor/models/task_modules/tracer/parsers.py b/mmrazor/models/task_modules/tracer/parsers.py
@@ -131,10 +131,10 @@ def parse_cat(tracer, grad_fn, module2name, param2module, cur_path,
         tracer.backward_trace(parent, module2name, param2module, Path(),
                               sub_path_list, visited, shared_module)
         sub_path_lists.append(sub_path_list)
-        cur_path.append(PathConcatNode(name, sub_path_lists))
+    cur_path.append(PathConcatNode(name, sub_path_lists))
 
-        result_paths.append(copy.deepcopy(cur_path))
-        cur_path.pop(-1)
+    result_paths.append(copy.deepcopy(cur_path))
+    cur_path.pop(-1)
 
 
 def parse_norm(tracer, grad_fn, module2name, param2module, cur_path,