timgaripov / swa
Stochastic Weight Averaging in PyTorch
☆971Updated 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
- 2.56%, 15.20%, 1.30% on CIFAR10, CIFAR100, and SVHN https://arxiv.org/abs/1708.04552☆551Updated 5 years ago
- Implementations of ideas from recent papers☆393Updated 4 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆986Updated 3 weeks ago
- Implementation of the mixup training method☆466Updated 6 years ago
- Standardizing weights to accelerate micro-batch training☆547Updated 3 years ago
- Efficient Learning of Augmentation Policy Schedules☆505Updated 5 years ago
- Over9000 optimizer☆426Updated 2 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆793Updated last year
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,601Updated 3 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Updated 4 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆335Updated 5 years ago
- ☆1,138Updated 2 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 11 months ago
- Fine-tune pretrained Convolutional Neural Networks with PyTorch☆721Updated 8 months ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,478Updated last year
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆609Updated 3 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)☆858Updated 2 years ago
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,041Updated 3 years ago
- Code for reproducing Manifold Mixup results (ICML 2019)☆489Updated last year
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆865Updated 4 years ago
- PyTorch Implementation for Deep Metric Learning Pipelines☆572Updated 4 years ago
- Wide Residual Networks (WideResNets) in PyTorch☆337Updated 3 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆881Updated 4 years ago
- High-level batteries-included neural network training library for Pytorch☆400Updated 3 years ago
- A lightweight library for PyTorch training tools and utilities☆1,688Updated last week
- Best CIFAR-10, CIFAR-100 results with wide-residual networks using PyTorch☆467Updated 5 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆759Updated 4 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆311Updated 5 years ago