moskomule / sam.pytorch
A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization
☆133Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for sam.pytorch
- This is the official implementation of TrivialAugment and a mini-library for the application of multiple image augmentation strategies in…☆150Updated last year
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆345Updated 9 months ago
- Differentiable Data Augmentation Library☆120Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆288Updated last year
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆196Updated last year
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆332Updated 3 years ago
- When Does Label Smoothing Help?_pytorch_implementationimp☆123Updated 4 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 3 years ago
- [Preprint] ConvMLP: Hierarchical Convolutional MLPs for Vision, 2021☆164Updated 2 years ago
- Official PyTorch implementation of the paper: "Solving ImageNet: a Unified Scheme for Training any Backbone to Top Results" (2022)☆192Updated last year
- [ICCV 2021] Influence-balanced Loss for Imbalanced Visual Classification☆99Updated 2 years ago
- Implementing Lambda Networks using Pytorch☆138Updated 3 years ago
- ☆565Updated 2 weeks ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 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)☆138Updated 2 years ago
- Implementation of Visual Transformer for Small-size Datasets☆117Updated 2 years ago
- A pytorch implementation of paper 'Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillation', …☆158Updated 2 years ago
- Official Pytorch implementation of MixMo framework☆84Updated 3 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆167Updated last year
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆158Updated 2 years ago
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 3 years ago
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆166Updated 2 years ago
- Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"☆426Updated last year
- [CVPR 2022] This repository includes the official project for the paper: TransMix: Attend to Mix for Vision Transformers.☆156Updated last year
- Code for the Convolutional Vision Transformer (ConViT)☆463Updated 3 years ago
- [TPAMI2022 & NeurIPS2020] Official implementation of Self-Adaptive Training☆127Updated 3 years ago
- Unofficial PyTorch implementation of "Meta Pseudo Labels"☆383Updated 9 months ago
- ☆96Updated last year
- PyTorch implementation of AutoAugment.☆157Updated 4 years ago