Unverified Commit d8a5aded authored by Super Daniel's avatar Super Daniel Committed by GitHub
Browse files

[hotfix] change namespace for meta_trace. (#1541)

parent a203b709
...@@ -20,7 +20,7 @@ def meta_trace(module: torch.nn.Module, *args, **kwargs) -> Graph: ...@@ -20,7 +20,7 @@ def meta_trace(module: torch.nn.Module, *args, **kwargs) -> Graph:
>>> graph.print_tabular() >>> graph.print_tabular()
""" """
graph = Graph() graph = Graph()
namespace = _Namespace() namespace = graph._graph_namespace
class MetaProxy(torch.Tensor): class MetaProxy(torch.Tensor):
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment