Stochastic Weight Averaging in PyTorch
☆982Aug 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,551Jul 31, 2021Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,110Apr 22, 2022Updated 4 years ago
- A PyTorch implementation of EfficientNet☆8,219Apr 8, 2022Updated 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,608Jun 16, 2021Updated 4 years ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,572Oct 9, 2025Updated 6 months ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,410Apr 12, 2024Updated 2 years ago
- 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
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 5 years ago
- ResNeSt: Split-Attention Networks☆3,263Dec 9, 2022Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A PyTorch implementation of shake-shake☆112Apr 21, 2020Updated 6 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,332Jul 8, 2025Updated 9 months ago
- ☆47Jan 11, 2021Updated 5 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,953Updated this week
- A CV toolkit for my papers.☆2,045Dec 21, 2024Updated last year
- Code for visualizing the loss landscape of neural nets☆3,175Apr 5, 2022Updated 4 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,817Sep 5, 2019Updated 6 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Jul 29, 2020Updated 5 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Nov 7, 2019Updated 6 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,168Mar 20, 2024Updated 2 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,986Apr 10, 2026Updated 3 weeks ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆339Mar 26, 2021Updated 5 years ago
- Dilated Residual Networks☆1,107Mar 9, 2020Updated 6 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆321Aug 4, 2022Updated 3 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☆692Nov 11, 2019Updated 6 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,289Jun 25, 2025Updated 10 months ago
- mixup: Beyond Empirical Risk Minimization☆1,196Oct 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,664Oct 8, 2020Updated 5 years ago
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆124Oct 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…☆403Jul 20, 2018Updated 7 years ago