Stochastic Weight Averaging in PyTorch
☆977Aug 1, 2021Updated 4 years ago
Alternatives and similar repositories for swa
Users that are interested in swa are comparing it to the libraries listed below
Sorting:
- Implementations of ideas from recent papers☆391Dec 22, 2020Updated 5 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆284Oct 24, 2022Updated 3 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆189Feb 3, 2019Updated 7 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Jul 31, 2021Updated 4 years ago
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆478Jul 6, 2023Updated 2 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,612Jun 16, 2021Updated 4 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,112Apr 22, 2022Updated 3 years ago
- A PyTorch implementation of EfficientNet☆8,221Apr 8, 2022Updated 3 years ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,409Apr 12, 2024Updated last year
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆870Jun 11, 2020Updated 5 years ago
- A PyTorch implementation of shake-shake☆111Apr 21, 2020Updated 5 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,335Jul 8, 2025Updated 7 months ago
- Official Pytorch implementation of CutMix regularizer☆1,247Sep 16, 2020Updated 5 years ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,567Oct 9, 2025Updated 4 months ago
- ResNeSt: Split-Attention Networks☆3,267Dec 9, 2022Updated 3 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- A CV toolkit for my papers.☆2,049Dec 21, 2024Updated last year
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,728Jul 6, 2023Updated 2 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,167Mar 20, 2024Updated last year
- An optimizer that trains as fast as Adam and as good as SGD.☆2,910Jul 23, 2023Updated 2 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,926Feb 24, 2026Updated last week
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years ago
- Code for visualizing the loss landscape of neural nets☆3,154Apr 5, 2022Updated 3 years ago
- Dilated Residual Networks☆1,104Mar 9, 2020Updated 5 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Nov 7, 2019Updated 6 years ago
- CondenseNet: Light weighted CNN for mobile devices☆691Nov 11, 2019Updated 6 years ago
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆124Oct 10, 2018Updated 7 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
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Jul 29, 2020Updated 5 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
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,268Jun 25, 2025Updated 8 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,989Feb 5, 2026Updated last month
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated 2 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆339Mar 26, 2021Updated 4 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,497May 16, 2023Updated 2 years ago
- 2.86% and 15.85% on CIFAR-10 and CIFAR-100☆297Oct 9, 2018Updated 7 years ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,583Jun 13, 2024Updated last year
- ☆47Jan 11, 2021Updated 5 years ago
- A state-of-the-art semi-supervised method for image recognition☆1,656Oct 8, 2020Updated 5 years ago