Jannoshh / simple-sam
Sharpness-Aware Minimization for Efficiently Improving Generalization
☆41Updated 3 years ago
Alternatives and similar repositories for simple-sam:
Users that are interested in simple-sam are comparing it to the libraries listed below
- ☆73Updated 2 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆56Updated 2 years ago
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆60Updated 3 years ago
- Keras implementation of Normalizer-Free Networks and SGD - Adaptive Gradient Clipping☆70Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 4 years ago
- ModelSoups for Tensorflow2 and Torch☆48Updated 2 years ago
- Implementation of self-supervised image-level contrastive pretraining methods using Keras.☆70Updated 3 years ago
- Cyclemoid implementation for PyTorch☆87Updated 3 years ago
- A hub for ResNet based models and pretrained weights in TensorFlow.☆21Updated 3 years ago
- PyTorch reimplementation of the Smooth ReLU activation function proposed in the paper "Real World Large Scale Recommendation Systems Repr…☆22Updated 2 years ago
- Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).☆126Updated 2 years ago
- Stochastic Weight Averaging Tutorials using pytorch.☆33Updated 4 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆212Updated 3 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 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)☆143Updated 2 years ago
- MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space☆40Updated 4 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆130Updated last week
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 3 years ago
- A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization☆136Updated 4 years ago
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 2 years ago
- ☆57Updated 2 years ago
- ☆14Updated 5 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆133Updated 4 years ago
- Learning to Initialize Neural Networks for Stable and Efficient Training☆138Updated 2 years ago
- Trains Transformer model variants. Data isn't shuffled between batches.☆141Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- GPT, but made only out of MLPs☆88Updated 3 years ago
- Layerwise Batch Entropy Regularization☆22Updated 2 years ago
- Keras Implementation of Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)☆96Updated 4 years ago