LucaBonfiglioli / nnvizLinks
Torch nn vizualization
☆52Updated last year
Alternatives and similar repositories for nnviz
Users that are interested in nnviz are comparing it to the libraries listed below
Sorting:
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆601Updated 5 months ago
- torchview: visualize pytorch models☆976Updated 2 months ago
- PhyCV: The First Physics-inspired Computer Vision Library☆519Updated 8 months ago
- Each week I create sketches covering key Computer Vision concepts. If you want to learn more about CV stick around!☆147Updated 2 years ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆428Updated 7 months ago
- A curated list of papers that released datasets along with their work☆126Updated 9 months ago
- Hiera: A fast, powerful, and simple hierarchical vision transformer.☆1,010Updated last year
- ☆76Updated last month
- Install PyTorch distributions with computation backend auto-detection☆254Updated 3 months ago
- Interactively inspect module inputs, outputs, parameters, and gradients.☆347Updated 2 months ago
- Self-Supervised Learning in PyTorch☆138Updated last year
- [ICML 2023] Official PyTorch implementation of Global Context Vision Transformers☆439Updated last year
- High-quality Neural Networks for Computer Vision 😎☆446Updated 2 years ago
- [ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmen…☆481Updated 2 years ago
- A Simplified PyTorch Implementation of Vision Transformer (ViT)☆196Updated last year
- [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention☆868Updated 2 weeks ago
- 👁️ + 💬 + 🎧 = 🤖 Curated list of top foundation and multimodal models! [Paper + Code + Examples + Tutorials]☆627Updated last year
- Tensors, for human consumption☆1,271Updated last month
- Cache PyTorch module outputs on-the-fly☆44Updated 3 months ago
- Documentation for Ross Wightman's timm image model library☆312Updated last year
- Computer Vision dataset analysis☆302Updated last year
- A distilled Segment Anything (SAM) model capable of running real-time with NVIDIA TensorRT☆784Updated last year
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆119Updated 5 months ago
- This repository is a curated collection of the most exciting and influential CVPR 2024 papers. 🔥 [Paper + Code + Demo]☆736Updated 2 months ago
- Probing the representations of Vision Transformers.☆328Updated 2 years ago
- A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the m…☆114Updated last year
- TensorDict is a pytorch dedicated tensor container.☆950Updated this week
- VisualTorch aims to help visualize Torch-based neural network architectures.☆94Updated last month
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆421Updated last year
- [NeurIPS'22] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training☆22Updated last year