simon-larsson / keras-swa
Simple stochastic weight averaging callback for Keras
☆63Updated 3 years ago
Alternatives and similar repositories for keras-swa:
Users that are interested in keras-swa are comparing it to the libraries listed below
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆60Updated 3 years ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 3 years ago
- Yet Another Neural Network Library 🤔☆26Updated 2 months ago
- https://www.kaggle.com/c/recursion-cellular-image-classification☆40Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Keras implementation of layer which performs augmentations of images using GPU.☆49Updated 3 years ago
- Implementation of modern data augmentation techniques in TensorFlow 2.x to be used in your training pipeline.☆34Updated 4 years ago
- In-place debugger for the fastai library and pytorch☆28Updated 3 years ago
- Fourth place solution to the "OpenVaccine: COVID-19 mRNA Vaccine Degradation Prediction" organized by Stanford University and Kaggle☆20Updated 4 years ago
- Comparing PyTorch Catalyst, Ignite, Lightning by sample code☆20Updated 2 years ago
- This library implements various metrics (including Kaggle Competition, Medicine) for evaluating ML, DL, AI models, and algorithms. 📐📊📈…☆14Updated 2 years ago
- Implementation of self-supervised image-level contrastive pretraining methods using Keras.☆70Updated 3 years ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- NeuralPy: A Keras like deep learning library works on top of PyTorch☆78Updated 9 months ago
- TF 2.x and PyTorch Lightning Callbacks for GPU monitoring☆92Updated 4 years ago
- Keras implementation of Cosine Annealing Scheduler☆44Updated 4 years ago
- This library is a location of the LegacyLogger for PyTorch Lightning.☆26Updated 2 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆112Updated 4 years ago
- Lookahead mechanism for optimizers in Keras.☆49Updated 3 years ago
- Minimalistic TensorFlow2+ deep metric/similarity learning library with loss functions, miners, and utils as embedding projector.☆37Updated 2 years ago
- Catalyst.Segmentation☆28Updated 3 years ago
- Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers☆167Updated 3 years ago
- Code for STFT Transformer used in BirdCLEF 2021 competition.☆78Updated 3 years ago
- A tiny Catalyst-like experiment runner framework on top of micrograd.☆51Updated 4 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆56Updated 2 years ago
- FasterAI: A repository for making smaller and faster models with the FastAI library.☆35Updated last year
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago