fix bug

open-mmlab · Sep 29, 2022 · 6a45316 · 6a45316
1 parent 62da5d8
commit 6a45316
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py b/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py
@@ -121,7 +121,7 @@ def test_replace_with_dynamic_ops(self):
             for unit_type in GROUPS:
                 with self.subTest(model=model_data, unit=unit_type):
                     model: nn.Module = model_data()
-                    graph = ModuleGraph.init_from_fx_tracer(model)
+                    graph = ModuleGraph.init_from_backward_tracer(model)
                     units: List[
                         MutableChannelUnit] = unit_type.init_from_graph(
                             graph)