← all repositories

luo3300612/Visualizer

A utility tool for extracting Attention maps from nested layers in PyTorch deep learning models.

1.3k stars Jupyter Notebook ML Frameworks
Visualizer
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

Visualizer is a developer tool that uses PyTorch hooks to extract Attention maps from deeply nested modules in Transformer-based models like Vision Transformer. It solves the common problem of having to trace through multiple layers of nested nn.Sequential or Block structures to retrieve intermediate attention outputs for visualization and debugging purposes.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.