moskomule / sam.pytorch
A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization
☆136Updated 3 years ago
Alternatives and similar repositories for sam.pytorch:
Users that are interested in sam.pytorch are comparing it to the libraries listed below
- Differentiable Data Augmentation Library☆121Updated 2 years ago
- This is the official implementation of TrivialAugment and a mini-library for the application of multiple image augmentation strategies in…☆153Updated last year
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆333Updated 3 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆159Updated 3 years ago
- When Does Label Smoothing Help?_pytorch_implementationimp☆126Updated 5 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆167Updated 2 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆347Updated last year
- ☆575Updated 3 months ago
- Unofficial PyTorch implementation of "Meta Pseudo Labels"☆387Updated last year
- Implementation of Online Label Smoothing in PyTorch☆95Updated 2 years ago
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆168Updated 2 years ago
- [ICCV 2021] Influence-balanced Loss for Imbalanced Visual Classification☆99Updated 2 years ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆139Updated 2 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆631Updated last year
- Official PyTorch implementation of "Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversity" (ICLR'21 Oral)☆103Updated 3 years ago
- ☆448Updated last year
- Official Pytorch implementation of MixMo framework☆84Updated 3 years ago
- SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data (AAAI 2021)☆129Updated 4 years ago
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆293Updated last year
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 4 years ago
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆197Updated 2 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆160Updated 3 years ago
- Official PyTorch implementation of the paper: "Solving ImageNet: a Unified Scheme for Training any Backbone to Top Results" (2022)☆191Updated 2 years ago
- ImageNet-R(endition) and DeepAugment (ICCV 2021)☆259Updated 3 years ago
- Implementation of Visual Transformer for Small-size Datasets☆119Updated 2 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 8 months ago
- ☆404Updated 3 years ago
- [TPAMI2022 & NeurIPS2020] Official implementation of Self-Adaptive Training☆129Updated 3 years ago
- PyTorch implementation of AutoAugment.☆158Updated 4 years ago