timgaripov / swa
Stochastic Weight Averaging in PyTorch
☆968Updated 3 years ago
Alternatives and similar repositories for swa:
Users that are interested in swa are comparing it to the libraries listed below
- Implementations of ideas from recent papers☆391Updated 4 years ago
- Over9000 optimizer☆426Updated 2 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,602Updated 3 years ago
- mixup: Beyond Empirical Risk Minimization☆1,172Updated 3 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆334Updated 5 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,478Updated last year
- 2.56%, 15.20%, 1.30% on CIFAR10, CIFAR100, and SVHN https://arxiv.org/abs/1708.04552☆549Updated 4 years ago
- A New Optimization Technique for Deep Neural Networks☆535Updated 3 years ago
- Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)☆853Updated 2 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆419Updated 9 months ago
- Implementation of the mixup training method☆465Updated 6 years ago
- Code for reproducing Manifold Mixup results (ICML 2019)☆487Updated 10 months ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆982Updated 6 months ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆790Updated last year
- Efficient Learning of Augmentation Policy Schedules☆505Updated 5 years ago
- ☆1,137Updated last year
- Standardizing weights to accelerate micro-batch training☆547Updated 2 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆592Updated 4 years ago
- Learning Confidence for Out-of-Distribution Detection in Neural Networks☆268Updated 6 years ago
- Visualization toolkit for neural networks in PyTorch! Demo -->☆738Updated last year
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆607Updated 3 years ago
- Public repo for Augmented Multiscale Deep InfoMax representation learning☆401Updated 4 years ago
- A large scale study of Knowledge Distillation.☆219Updated 4 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆631Updated last year
- Fine-tune pretrained Convolutional Neural Networks with PyTorch☆723Updated 6 months ago
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,038Updated 3 years ago
- High-level batteries-included neural network training library for Pytorch☆398Updated 3 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆309Updated 5 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆865Updated 4 years ago
- Wide Residual Networks (WideResNets) in PyTorch☆337Updated 3 years ago