moskomule / shampoo.pytorch
An implementation of shampoo
☆74Updated 6 years ago
Alternatives and similar repositories for shampoo.pytorch:
Users that are interested in shampoo.pytorch are comparing it to the libraries listed below
- This repository is no longer maintained. Check☆81Updated 4 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 4 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆148Updated 7 years ago
- ☆34Updated 6 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 4 years ago
- The Singular Values of Convolutional Layers☆71Updated 6 years ago
- A PyTorch implementation of shake-shake☆111Updated 4 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 6 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆83Updated 6 years ago
- Odds and Ends and Things I've implemented.☆78Updated 5 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated last year
- An Implementation of "Small steps and giant leaps: Minimal Newton solvers for Deep Learning" In pytorch☆21Updated 6 years ago
- ☆23Updated 6 years ago
- Implementation of soft parameter sharing for neural networks☆69Updated 4 years ago
- [NeurIPS'19] [PyTorch] Adaptive Regularization in NN☆67Updated 5 years ago
- Code for ICLR2018 paper: Improving GAN Training via Binarized Representation Entropy (BRE) Regularization - Y. Cao · W Ding · Y.C. Lui · …☆20Updated 6 years ago
- Structured Bayesian Pruning, NIPS 2017☆73Updated 4 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- TensorFlow implementation of (Momentum) Stochastic Variance-Adapted Gradient.☆44Updated 6 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- ☆34Updated 5 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated last year
- Generalized Framework for PyTorch☆32Updated 3 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- Implementation of the Deep Frank-Wolfe Algorithm -- Pytorch☆62Updated 3 years ago
- ☆61Updated last year
- ☆21Updated 4 years ago
- Implementation of "Variational Dropout and the Local Reparameterization Trick" paper with Pytorch☆50Updated 7 years ago