sayakpaul / Sharpness-Aware-Minimization-TensorFlowLinks
Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.
☆61Updated 4 years ago
Alternatives and similar repositories for Sharpness-Aware-Minimization-TensorFlow
Users that are interested in Sharpness-Aware-Minimization-TensorFlow are comparing it to the libraries listed below
Sorting:
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆59Updated 3 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆86Updated 4 years ago
- Implementation of modern data augmentation techniques in TensorFlow 2.x to be used in your training pipeline.☆34Updated 5 years ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 4 years ago
- Official code for the Stochastic Polyak step-size optimizer☆139Updated last year
- Minimal implementation of SimSiam (https://arxiv.org/abs/2011.10566) in TensorFlow 2.☆98Updated 4 years ago
- Python implementation of GLN in different frameworks☆96Updated 5 years ago
- ☆79Updated 5 years ago
- Cyclemoid implementation for PyTorch