LucaBonfiglioli / nnviz
Torch nn vizualization
☆50Updated last year
Alternatives and similar repositories for nnviz:
Users that are interested in nnviz are comparing it to the libraries listed below
- Timm model explorer☆39Updated last year
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆577Updated 2 months ago
- ☆69Updated last month
- ☆65Updated 6 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆424Updated 4 months ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆115Updated 2 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
- A curated list of papers that released datasets along with their work☆125Updated 6 months ago
- Self-Supervised Learning in PyTorch☆137Updated last year
- [ICML 2023] Official PyTorch implementation of Global Context Vision Transformers☆432Updated last year
- Large Scale Non maximum Suppression. This project implements a O(nlog(n)) approach for non max suppression, useful for object detection r…☆68Updated 6 months ago
- Simplify Your Visual Data Ops. Find and visualize issues with your computer vision datasets such as duplicates, anomalies, data leakage, …☆69Updated this week
- ☆201Updated last year
- A tutorial introducing knowledge distillation as an optimization technique for deployment on NVIDIA Jetson☆192Updated last year
- torchview: visualize pytorch models☆918Updated 2 weeks ago
- vision transformers with pytorch and pytorch lightningUpdated 6 months ago
- Install PyTorch distributions with computation backend auto-detection☆251Updated last week
- RobustSAM: Segment Anything Robustly on Degraded Images (CVPR 2024 Highlight)☆348Updated 8 months ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆40Updated 2 months ago
- Contains the "pycocotools" package on PyPI. Changes made to the official cocoapi about packaging.☆142Updated 8 months ago
- Continuation of an abandoned project fast-coco-eval☆100Updated 3 months ago
- Documentation, notes, links, etc for streams.☆78Updated last year
- ☆150Updated 8 months ago
- ☆131Updated 2 years ago
- NumPy arrays, ready for human consumption☆69Updated 9 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆183Updated 7 months ago
- A multi-backend (TensorFlow, PyTorch, JAX, and NumPy) implementation of the Segment Anything model in Keras 3.0☆32Updated last year
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆185Updated 2 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 9 months ago
- Probing the representations of Vision Transformers.☆324Updated 2 years ago