tuomaso / train_mnist_fastLinks
How to train a CNN to 99% accuracy on MNIST in less than a second on a laptop
☆70Updated last year
Alternatives and similar repositories for train_mnist_fast
Users that are interested in train_mnist_fast are comparing it to the libraries listed below
Sorting:
- Learning with Signatures☆57Updated 3 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated last year
- Yet Another Neural Network Library 🤔☆27Updated 3 months ago
- A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).☆23Updated 4 years ago
- Pytorch cyclic cosine decay learning rate scheduler☆49Updated 4 years ago
- Figures I made during my PhD in Deep Learning, for my models and for context☆84Updated 4 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- Create animations for the optimization trajectory of neural nets☆161Updated last year
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- ☆133Updated 2 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆218Updated 4 years ago
- Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"☆31Updated 5 years ago
- Adversarial examples to the new ConvNeXt architecture☆20Updated 3 years ago
- A survey of all the 'X is all you need'☆75Updated 4 years ago
- Convert scikit-learn models to PyTorch modules☆169Updated last year
- PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021)☆31Updated 4 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Updated last year
- ☆35Updated 5 years ago
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆88Updated 4 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- Official PyTorch implementation of LilNetX: Lightweight Networks with EXtreme Model Compression and Structured Sparsification☆47Updated 3 years ago
- Large dataset storage format for Pytorch☆45Updated 4 years ago
- ☆37Updated 3 years ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆16Updated 4 years ago