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

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.