Tutorials
Step-by-step guide to help you get the most out of TorchVista
1
→
2
→
3
→
4
→
5
→
6
→
7
→
Basic Usage
Learn how to visualize your first PyTorch model with trace_model
Exploring Nested Modules
Expand and collapse module hierarchies to explore complex architectures
Forced module tracing depth
Trace any module up to the desired nesting depth
Collapse modules after depth
Control which modules are initially expanded in the visualization
Hiding non-gradient nodes
Simplify the visualization by hiding constant tensors and scalars
Naming Input/Output tensors with nested dicts
Use nested dictionary inputs and outputs to name tensor nodes
Compressed view for repeated structures
Collapse repeated layers into a single representation