MisaOgura / flashtorch

Visualization toolkit for neural networks in PyTorch! Demo -->
https://youtu.be/18Iw4qYqfPo
MIT License
734 stars 87 forks source link

[Q] Does this support Vision Transformers (ViT)? #47

Open NightMachinery opened 6 months ago

NightMachinery commented 6 months ago

Does this support Vision Transformers (ViT)? Do you know of any repos or papers on feature visualization (activation maximization) on ViTs?