- I made an open-source Python package, TorchLens, that can visualize the structure of any PyTorch model and extract any intermediate activations you want in one line of code. https://github.com/johnmarktaylor91/torchlens 2 comments python
Linking pages
Linked pages
- [2103.14030] Swin Transformer: Hierarchical Vision Transformer using Shifted Windows https://arxiv.org/abs/2103.14030 20 comments
- GitHub - archinetai/surgeon-pytorch: A library to inspect and extract intermediate layers of PyTorch models. https://github.com/archinetai/surgeon-pytorch 13 comments
- Extracting and visualizing hidden activations and computational graphs of PyTorch models with TorchLens | Scientific Reports https://www.nature.com/articles/s41598-023-40807-0 0 comments
Would you like to stay up to date with Python? Checkout Python
Weekly.