timgaripov / swa
Stochastic Weight Averaging in PyTorch
☆972Updated 3 years ago
Alternatives and similar repositories for swa:
Users that are interested in swa are comparing it to the libraries listed below
- mixup: Beyond Empirical Risk Minimization☆1,178Updated 3 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆988Updated 3 weeks ago
- Implementations of ideas from recent papers☆393Updated 4 years ago
- A New Optimization Technique for Deep Neural Networks☆535Updated 3 years ago
- Standardizing weights to accelerate micro-batch training☆547Updated 3 years ago
- Implementation of the mixup training method☆466Updated 6 years ago
- Efficient Learning of Augmentation Policy Schedules☆506Updated 5 years ago
- 2.56%, 15.20%, 1.30% on CIFAR10, CIFAR100, and SVHN https://arxiv.org/abs/1708.04552☆550Updated 5 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,599Updated 3 years ago
- ☆1,139Updated 2 years ago
- Code snippets created for the PyTorch discussion board☆561Updated 4 years ago
- Over9000 optimizer☆426Updated 2 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆335Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆312Updated 5 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Updated 4 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆989Updated 7 months ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆797Updated last year
- Best CIFAR-10, CIFAR-100 results with wide-residual networks using PyTorch☆469Updated 5 years ago
- Learning Confidence for Out-of-Distribution Detection in Neural Networks☆274Updated 7 years ago
- Code for reproducing Manifold Mixup results (ICML 2019)☆491Updated last year
- High-level batteries-included neural network training library for Pytorch☆401Updated 3 years ago
- Visualization toolkit for neural networks in PyTorch! Demo -->☆741Updated last year
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,479Updated last year
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆609Updated 3 years ago
- Scaling and Benchmarking Self-Supervised Visual Representation Learning☆586Updated 3 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆420Updated 11 months ago
- Public repo for Augmented Multiscale Deep InfoMax representation learning☆400Updated 4 years ago
- Open source release of the evaluation benchmark suite described in "Realistic Evaluation of Deep Semi-Supervised Learning Algorithms"☆460Updated 5 years ago
- Wide Residual Networks (WideResNets) in PyTorch☆338Updated 4 years ago
- Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)☆861Updated 2 years ago