search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

Show more custom information of tensor and gradient #82

Open GIS-PuppetMaster opened 9 months ago

GIS-PuppetMaster commented 9 months ago
  1. add support to show multiple lines attribute value
  2. allowed to customize which attr of the variable can be displayed
GIS-PuppetMaster commented 8 months ago

This is only a demo version, it seems Graphviz always displays the attribute with center alignment since this tool uses caption to contain attributes. Better to display them with left alignment [TODO]