Stochastic Weight Averaging in PyTorch
☆983Aug 1, 2021Updated 5 years ago
Alternatives and similar repositories for swa
Users that are interested in swa are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementations of ideas from recent papers☆390Dec 22, 2020Updated 5 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆286Oct 24, 2022Updated 3 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆190Feb 3, 2019Updated 7 years ago
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆477Jul 6, 2023Updated 3 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,547Jul 31, 2021Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,102Apr 22, 2022Updated 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,604Jun 16, 2021Updated 5 years ago
- A PyTorch implementation of EfficientNet☆8,220Apr 8, 2022Updated 4 years ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,575Oct 9, 2025Updated last year
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,409Apr 12, 2024Updated 2 years ago
- Official Pytorch implementation of CutMix regularizer☆1,251Sep 16, 2020Updated 6 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆871Jun 11, 2020Updated 6 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,501Apr 8, 2021Updated 5 years ago
- ResNeSt: Split-Attention Networks☆3,260Sep 11, 2026Updated last month
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- A PyTorch implementation of shake-shake☆111Apr 21, 2020Updated 6 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆9,003Updated this week
- ☆47Jan 11, 2021Updated 5 years ago
- A CV toolkit for my papers.☆2,044Sep 11, 2026Updated last month
- Code for visualizing the loss landscape of neural nets☆3,206Apr 5, 2022Updated 4 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆338Mar 26, 2021Updated 5 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,904Jul 23, 2023Updated 3 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,807Sep 5, 2019Updated 7 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,161Mar 20, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆595Jul 29, 2020Updated 6 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆321Aug 4, 2022Updated 4 years ago
- Dilated Residual Networks☆1,102Mar 9, 2020Updated 6 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Nov 7, 2019Updated 6 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆8,002Jul 14, 2026Updated 3 months ago
- Training Very Deep Neural Networks Without Skip-Connections☆591Jun 9, 2018Updated 8 years ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,722Jul 6, 2023Updated 3 years ago
- CondenseNet: Light weighted CNN for mobile devices☆690Nov 11, 2019Updated 6 years ago
- mixup: Beyond Empirical Risk Minimization☆1,195Oct 12, 2021Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Accelerate Neural Net Training by Progressively Freezing Layers☆213Aug 15, 2018Updated 8 years ago
- A state-of-the-art semi-supervised method for image recognition☆1,682Oct 8, 2020Updated 6 years ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,308Jun 25, 2025Updated last year
- Image Test Time Augmentation with PyTorch!☆1,032Jul 28, 2023Updated 3 years ago
- Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' pap…☆404Jul 20, 2018Updated 8 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,050Mar 2, 2024Updated 2 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,493May 16, 2023Updated 3 years ago