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
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆344Updated last year
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆197Updated 2 years ago
- Differentiable Data Augmentation Library☆123Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- When Does Label Smoothing Help?_pytorch_implementationimp☆125Updated 5 years ago
- Official Pytorch implementation of MixMo framework☆83Updated 3 years ago
- This is the official implementation of TrivialAugment and a mini-library for the application of multiple image augmentation strategies in…☆156Updated last year
- [Preprint] ConvMLP: Hierarchical Convolutional MLPs for Vision, 2021☆166Updated 2 years ago
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)