tuomaso / train_mnist_fast
How to train a CNN to 99% accuracy on MNIST in less than a second on a laptop
☆60Updated 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
- Learning with Signatures☆56Updated 2 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.☆129Updated 4 months ago
- ☆73Updated 2 years ago
- A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).☆23Updated 3 years ago
- ☆16Updated last year
- Parameterizing Mixing Links in Sparse Factors Works Better than Dot-Product Self-Attention (CVPR 2022)☆20Updated 2 years ago
- Pytorch cyclic cosine decay learning rate scheduler☆48Updated 3 years ago
- Implementing DropPath/StochasticDepth in PyTorch☆16Updated 3 years ago
- A database of over 1.4 billion 3x3 convolution filters extracted from hundreds of diverse CNN models with relevant meta information (CVPR…☆32Updated last year
- A "gym" style toolkit for building lightweight NAS systems.☆13Updated 2 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- ☆36Updated 3 years ago
- [CogSci'21] Study of human inductive biases in CNNs and Transformers.☆43Updated 3 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- Layerwise Batch Entropy Regularization☆22Updated 2 years ago
- Codebase for the paper "Beyond BatchNorm: Towards a Unified Understanding of Normalization in Deep Learning"☆17Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Comparing four automatic image augmentation techniques in PyTorch: AutoAugment, RandAugment, AugMix, and TrivialAugment☆28Updated 2 years ago
- A crash course on PyTorch hooks☆40Updated 4 years ago
- Implementation of Spectral Leakage and Rethinking the Kernel Size in CNNs in Pytorch☆14Updated 4 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆21Updated last year
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆222Updated last month
- Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).☆126Updated 2 years ago
- Experiments with the ideas presented in https://arxiv.org/abs/2003.00152 by Frankle et al.☆29Updated 4 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021)☆30Updated 3 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Various test models in WNNX format. It can view with `pip install wnetron && wnetron`☆12Updated 2 years ago
- Create animations for the optimization trajectory of neural nets☆147Updated last year
- PyTorch and Torch implementation for our accepted CVPR 2020 paper (Oral): Controllable Orthogonalization in Training DNNs☆24Updated 4 years ago