Swall0w / torchstat
Model analyzer in PyTorch
☆1,482Updated 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 neural networks in pytorch framework☆2,880Updated 2 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,977Updated 9 months ago
- Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)☆1,515Updated 4 years ago
- A CV toolkit for my papers.☆2,048Updated 3 months ago
- My best practice of training large dataset using PyTorch.☆1,096Updated 11 months ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆882Updated 4 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,152Updated last year
- RepVGG: Making VGG-style ConvNets Great Again☆3,381Updated 2 years ago
- A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility☆1,928Updated 2 years ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,570Updated 10 months ago
- ResNeSt: Split-Attention Networks☆3,253Updated 2 years ago
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,390Updated 5 years ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,387Updated last year
- ☆668Updated 3 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,037Updated last year
- ☆1,495Updated 4 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,478Updated last year
- Some tricks of pytorch...☆1,185Updated 9 months ago
- PyTorch implementation of SENet☆2,317Updated 4 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,500Updated 4 years ago
- Collection of recent methods on (deep) neural network compression and acceleration.☆943Updated last week
- Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019☆916Updated 2 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,107Updated 4 months ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,454Updated 6 years ago
- A curated list of neural network pruning resources.☆2,435Updated last year
- PyTorch layer-by-layer model profiler☆606Updated 3 years ago
- ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks☆841Updated 2 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 6 months ago
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,307Updated 3 years ago
- Network Slimming (Pytorch) (ICCV 2017)☆917Updated 4 years ago