moskomule / sam.pytorch
A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization
☆135Updated 4 years ago
Alternatives and similar repositories for sam.pytorch:
Users that are interested in sam.pytorch are comparing it to the libraries listed below
- This is the official implementation of TrivialAugment and a mini-library for the application of multiple image augmentation strategies in…☆156Updated 2 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆334Updated 4 years ago
- Differentiable Data Augmentation Library☆123Updated 2 years ago
- When Does Label Smoothing Help?_pytorch_implementationimp☆125Updated 5 years ago
- ☆595Updated 5 months ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 4 years ago
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆197Updated 2 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆345Updated last year
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆143Updated 2 years ago
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 4 years ago
- [CVPR 2021] Code for "Augmentation Strategies for Learning with Noisy Labels".☆113Updated 3 years ago
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆159Updated 3 years ago
- [ICCV 2021] Influence-balanced Loss for Imbalanced Visual Classification☆100Updated 3 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆295Updated last year
- ☆138Updated 3 years ago
- [TPAMI2022 & NeurIPS2020] Official implementation of Self-Adaptive Training☆129Updated 3 years ago
- Official Pytorch implementation of MixMo framework☆83Updated 3 years ago
- [NeurIPS 2021] Official codes for "Efficient Training of Visual Transformers with Small Datasets".☆141Updated 3 months ago
- [Preprint] ConvMLP: Hierarchical Convolutional MLPs for Vision, 2021☆166Updated 2 years ago
- PyTorch implementation of AutoAugment.☆159Updated 4 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆168Updated 2 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 11 months ago
- Official PyTorch implementation of the paper: "Solving ImageNet: a Unified Scheme for Training any Backbone to Top Results" (2022)☆191Updated 2 years ago
- Official PyTorch implementation of "Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversity" (ICLR'21 Oral)☆103Updated 3 years ago
- Official Repsoitory for "Activate or Not: Learning Customized Activation." [CVPR 2021]☆204Updated 3 years ago
- Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch☆425Updated 3 years ago
- Pytorch Implementation of Deep Networks with Stochastic Depth☆63Updated 6 years ago
- Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"☆427Updated last year