Model analyzer in PyTorch
☆1,501Mar 19, 2023Updated 2 years ago
Alternatives and similar repositories for torchstat
Users that are interested in torchstat are comparing it to the libraries listed below
Sorting:
- Count the MACs / FLOPs of your PyTorch model.☆5,082Jul 8, 2024Updated last year
- Flops counter for neural networks in pytorch framework☆2,975Aug 20, 2025Updated 6 months ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated 2 years ago
- ☆1,510Aug 27, 2020Updated 5 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,926Updated this week
- Implementations of ideas from recent papers☆391Dec 22, 2020Updated 5 years ago
- A CV toolkit for my papers.☆2,049Dec 21, 2024Updated last year
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,114Apr 22, 2022Updated 3 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,822Sep 5, 2019Updated 6 years ago
- Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)☆1,516Jun 7, 2020Updated 5 years ago
- A PyTorch implementation of EfficientNet☆8,221Apr 8, 2022Updated 3 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,224Jan 12, 2026Updated last month
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,989Feb 5, 2026Updated 3 weeks ago
- TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision☆2,257Nov 19, 2020Updated 5 years ago
- PyTorch code for the "Deep Neural Networks with Box Convolutions" paper☆510Jan 20, 2020Updated 6 years ago
- Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019☆928Mar 9, 2023Updated 2 years ago
- Pytorch implementation of newly added convolution☆584Oct 26, 2019Updated 6 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,911Jul 23, 2023Updated 2 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,335Jul 8, 2025Updated 7 months ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,386Feb 16, 2023Updated 3 years ago
- Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset☆5,061Jan 15, 2024Updated 2 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,637Updated this week
- Codebase for Image Classification Research, written in PyTorch.☆2,168Mar 20, 2024Updated last year
- Submission to https://pytorch.org/hub/☆1,432Apr 15, 2024Updated last year
- Deformable Convolutional Networks☆4,116Sep 27, 2021Updated 4 years ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,347Dec 9, 2023Updated 2 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,229Dec 29, 2021Updated 4 years ago
- Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' pap…☆402Jul 20, 2018Updated 7 years ago
- A PyTorch implementation of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.☆275Aug 15, 2018Updated 7 years ago
- Non-local Neural Networks for Video Classification☆1,996Sep 15, 2021Updated 4 years ago
- Detectorch - detectron for PyTorch☆559Oct 30, 2018Updated 7 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,749Updated this week
- TorchCV: a PyTorch vision library mimics ChainerCV☆396Dec 25, 2018Updated 7 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆870Jun 11, 2020Updated 5 years ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,397Feb 23, 2026Updated last week
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,481Mar 19, 2021Updated 4 years ago
- LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)☆724Feb 23, 2020Updated 6 years ago