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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementations of ideas from recent papers☆391Dec 22, 2020Updated 5 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆285Oct 24, 2022Updated 3 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆188Feb 3, 2019Updated 7 years ago
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆477Jul 6, 2023Updated 2 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Jul 31, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,109Apr 22, 2022Updated 3 years ago
- A PyTorch implementation of EfficientNet☆8,219Apr 8, 2022Updated 3 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,610Jun 16, 2021Updated 4 years ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,568Oct 9, 2025Updated 5 months ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,410Apr 12, 2024Updated last year
- Official Pytorch implementation of CutMix regularizer☆1,246Sep 16, 2020Updated 5 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆869Jun 11, 2020Updated 5 years ago
- ResNeSt: Split-Attention Networks☆3,264Dec 9, 2022Updated 3 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 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.
- A PyTorch implementation of shake-shake☆112Apr 21, 2020Updated 5 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,334Jul 8, 2025Updated 8 months ago
- ☆47Jan 11, 2021Updated 5 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,936Mar 17, 2026Updated last week
- A CV toolkit for my papers.☆2,048Dec 21, 2024Updated last year
- Code for visualizing the loss landscape of neural nets☆3,158Apr 5, 2022Updated 3 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,907Jul 23, 2023Updated 2 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,818Sep 5, 2019Updated 6 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Jul 29, 2020Updated 5 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Nov 7, 2019Updated 6 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,166Mar 20, 2024Updated 2 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,990Feb 5, 2026Updated last month
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆339Mar 26, 2021Updated 4 years ago
- Dilated Residual Networks☆1,104Mar 9, 2020Updated 6 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,725Jul 6, 2023Updated 2 years ago
- CondenseNet: Light weighted CNN for mobile devices☆691Nov 11, 2019Updated 6 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,286Jun 25, 2025Updated 9 months ago
- mixup: Beyond Empirical Risk Minimization☆1,199Oct 12, 2021Updated 4 years ago
- Image Test Time Augmentation with PyTorch!☆1,028Jul 28, 2023Updated 2 years ago
- A state-of-the-art semi-supervised method for image recognition☆1,659Oct 8, 2020Updated 5 years ago
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆123Oct 10, 2018Updated 7 years ago
- Accelerate Neural Net Training by Progressively Freezing Layers☆211Aug 15, 2018Updated 7 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