Swall0w / torchstat
Model analyzer in PyTorch
☆1,477Updated 2 years ago
Alternatives and similar repositories for torchstat:
Users that are interested in torchstat are comparing it to the libraries listed below
- Flops counter for convolutional networks in pytorch framework☆2,869Updated 2 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,964Updated 8 months ago
- Codebase for Image Classification Research, written in PyTorch.☆2,150Updated last year
- My best practice of training large dataset using PyTorch.☆1,092Updated 10 months ago
- A CV toolkit for my papers.☆2,047Updated 3 months ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆880Updated 4 years ago
- PyTorch implementation of SENet☆2,313Updated 4 years ago
- ResNeSt: Split-Attention Networks☆3,246Updated 2 years ago
- ☆1,493Updated 4 years ago
- Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)☆1,516Updated 4 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,459Updated 2 years ago
- Code for paper " AdderNet: Do We Really Need Multiplications in Deep Learning?"☆958Updated 3 years ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,205Updated 4 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,499Updated 3 years ago
- Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019☆917Updated 2 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,376Updated 2 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 5 months ago
- Some tricks of pytorch...☆1,181Updated 9 months ago
- [ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment☆1,902Updated last year
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,452Updated 6 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,093Updated 3 months ago
- A quickstart and benchmark for pytorch distributed training.☆1,657Updated 7 months ago
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,309Updated 3 years ago
- Debug PyTorch code using PySnooper☆799Updated 3 years ago
- (ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"☆1,088Updated 2 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,480Updated last year
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,568Updated 9 months ago
- Network Slimming (Pytorch) (ICCV 2017)☆916Updated 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,602Updated 3 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,033Updated last year