moskomule / sam.pytorch
A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization
☆134Updated 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…☆152Updated last year
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆345Updated 10 months ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- Differentiable Data Augmentation Library☆120Updated 2 years ago
- ☆566Updated 3 weeks ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆332Updated 3 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- Implementation of Visual Transformer for Small-size Datasets☆117Updated 2 years ago
- When Does Label Smoothing Help?_pytorch_implementationimp☆124Updated 4 years ago
- [ICCV 2021] Influence-balanced Loss for Imbalanced Visual Classification☆99Updated 2 years ago
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆196Updated last year
- 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
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 3 years ago
- Unofficial PyTorch implementation of "Meta Pseudo Labels"☆383Updated 10 months ago
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆288Updated last year
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆158Updated 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
- NeurIPS 2021, Official codes for "Efficient Training of Visual Transformers with Small Datasets".☆139Updated last year
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆166Updated 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
- A pytorch implementation of paper 'Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillation', …☆158Updated 2 years ago
- MUXConv: Information Multiplexing in Convolutional Neural Networks (CVPR '20), Pytorch Implementation☆51Updated 3 years ago
- Official Pytorch implementation of MixMo framework☆84Updated 3 years ago
- Implementing Lambda Networks using Pytorch☆138Updated 3 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆167Updated 2 years ago
- Code for ICML 2022 paper "Out-of-distribution Detection with Deep Nearest Neighbors"☆175Updated 4 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆406Updated last month
- Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"☆425Updated last year
- Official Repsoitory for "Activate or Not: Learning Customized Activation." [CVPR 2021]☆203Updated 3 years ago