Tutorials

1

Basic Usage

Learn how to visualize your first PyTorch model with trace_model

→ 2

Exploring Nested Modules

Expand and collapse module hierarchies to explore complex architectures

→ 3

Forced module tracing depth

Trace any module up to the desired nesting depth

→ 4

Collapse modules after depth

Control which modules are initially expanded in the visualization

→ 5

Hiding non-gradient nodes

Simplify the visualization by hiding constant tensors and scalars

→ 6

Naming Input/Output tensors with nested dicts

Use nested dictionary inputs and outputs to name tensor nodes

→ 7

Compressed view for repeated structures

Collapse repeated layers into a single representation

→