pytorch
d579838e - [torch][fx] Add ignore_parameters_and_buffers kwarg to FxGraphDrawer (#79982)

Commit
2 years ago
[torch][fx] Add ignore_parameters_and_buffers kwarg to FxGraphDrawer (#79982) Summary: Add an `ignore_parameters_and_buffers` parameter which will tell the graph drawer to leave off adding parameter and buffer nodes in the dot graph. This is useful for large networks, where we want to view the graph to get an idea of the topology and the shapes without needing to see every detail. Removing these buffers de-clutters the graph significantly without detracting much information. Reviewed By: jfix71 Differential Revision: D37317917 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79982 Approved by: https://github.com/jfix71
Author
Riley Dulin
Committer
Parents
Loading