simon-larsson / keras-swa
Simple stochastic weight averaging callback for Keras
☆63Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for keras-swa
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆60Updated 3 years ago
- Keras implementation of Cosine Annealing Scheduler☆44Updated 4 years ago
- Lookahead mechanism for optimizers in Keras.☆49Updated 3 years ago
- Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers☆167Updated 2 years ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 3 years ago
- Keras implementation of layer which performs augmentations of images using GPU.☆49Updated 3 years ago
- This library is a location of the LegacyLogger for PyTorch Lightning.☆27Updated 2 years ago
- Keras callbacks for one-cycle training, cyclic learning rate (CLR) training, and learning rate range test.☆47Updated 5 years ago
- Implementation of self-supervised image-level contrastive pretraining methods using Keras.☆69Updated 3 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆80Updated 3 years ago
- TF 2.x and PyTorch Lightning Callbacks for GPU monitoring☆92Updated 4 years ago
- Fourth place solution to the "OpenVaccine: COVID-19 mRNA Vaccine Degradation Prediction" organized by Stanford University and Kaggle☆20Updated 4 years ago
- NeuralPy: A Keras like deep learning library works on top of PyTorch☆79Updated 5 months ago
- ☆86Updated 3 years ago
- https://www.kaggle.com/c/recursion-cellular-image-classification☆40Updated last year
- Catalyst.Segmentation☆28Updated 3 years ago
- Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.☆45Updated 3 years ago
- ☆26Updated 5 years ago
- ☆71Updated 3 years ago
- This library implements various metrics (including Kaggle Competition, Medicine) for evaluating ML, DL, AI models, and algorithms. 📐📊📈…☆14Updated last year
- In-place debugger for the fastai library and pytorch☆28Updated 3 years ago
- Semantic Segmentation with Pytorch-Lightning☆63Updated 3 years ago
- ☆47Updated 3 years ago
- Keras callback function for stochastic weight averaging☆55Updated 2 years ago
- Comparing PyTorch Catalyst, Ignite, Lightning by sample code☆20Updated last year
- AdamW optimizer for Keras☆114Updated 5 years ago
- TF2.0 port for Augmix paper☆79Updated 4 years ago