Swall0w / torchstat
Model analyzer in PyTorch
☆1,471Updated last year
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,848Updated 3 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,940Updated 6 months ago
- My best practice of training large dataset using PyTorch.☆1,093Updated 8 months ago
- A CV toolkit for my papers.☆2,047Updated 3 weeks ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆880Updated 4 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,143Updated 9 months ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,501Updated 3 years ago
- ☆1,486Updated 4 years ago
- Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)☆1,511Updated 4 years ago
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,385Updated 5 years ago
- A memory-efficient implementation of DenseNets☆1,525Updated last year
- Classification with PyTorch.☆1,706Updated 7 months ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,568Updated 7 months ago
- pytorch memory track code☆1,012Updated 3 years ago
- PyTorch implementation of SENet☆2,296Updated 3 years ago
- A lightweight library for PyTorch training tools and utilities☆1,678Updated last week
- Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019☆914Updated last year
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,450Updated 6 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆982Updated 3 months ago
- Network Slimming (Pytorch) (ICCV 2017)☆914Updated 4 years ago
- Some tricks of pytorch...☆1,177Updated 6 months ago
- pip install antialiased-cnns to improve stability and accuracy☆1,659Updated 9 months ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,198Updated 3 years ago
- ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks☆838Updated last year
- ResNeSt: Split-Attention Networks☆3,248Updated 2 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,455Updated 2 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,360Updated last year
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,474Updated last year
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆867Updated 4 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,057Updated last month