simon-larsson / keras-swaLinks
Simple stochastic weight averaging callback for Keras
☆63Updated 4 years ago
Alternatives and similar repositories for keras-swa
Users that are interested in keras-swa are comparing it to the libraries listed below
Sorting:
- Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers☆168Updated 3 years ago
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆61Updated 3 years ago
- NeuralPy: A Keras like deep learning library works on top of PyTorch☆79Updated last year
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Keras implementation of Cosine Annealing Scheduler☆44Updated 5 years ago
- TF 2.x and PyTorch Lightning Callbacks for GPU monitoring☆92Updated 5 years ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 4 years ago
- Code for the CVPR 2019 paper : Spectral Metric for Dataset Complexity Assessment☆45Updated last year
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- Fourth place solution to the "OpenVaccine: COVID-19 mRNA Vaccine Degradation Prediction" organized by Stanford University and Kaggle☆20Updated 5 years ago
- Keras callbacks for one-cycle training, cyclic learning rate (CLR) training, and learning rate range test.☆47Updated 6 years ago
- 🎲 Iterable dataset resampling in PyTorch☆92Updated 3 years ago
- Yet Another Neural Network Library 🤔☆27Updated last month
- Tensorflow Keras implementation of ordinal regression using consistent rank logits (CORAL) by Cao et al. (2019)☆81Updated last month
- ☆71Updated 4 years ago
- ☆26Updated 6 years ago
- Creates a learning-curve plot for Jupyter/Colab notebooks that is updated in real-time.☆177Updated 3 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated 2 months ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆57Updated 3 years ago
- Comparing PyTorch Catalyst, Ignite, Lightning by sample code☆20Updated 2 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- Python implementation of GLN in different frameworks☆97Updated 5 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆217Updated 4 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Tensorflow implementation of a linear attention architecture☆44Updated 4 years ago
- Implementation of Rectified Adam in Keras☆70Updated 6 years ago
- Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf☆147Updated 3 years ago
- Semantic Segmentation with Pytorch-Lightning☆63Updated 4 years ago
- Implementation of self-supervised image-level contrastive pretraining methods using Keras.☆70Updated 4 years ago
- Minimalistic TensorFlow2+ deep metric/similarity learning library with loss functions, miners, and utils as embedding projector.☆38Updated 2 years ago