keskarnitish / large-batch-trainingLinks
Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"
☆139Updated 8 years ago
Alternatives and similar repositories for large-batch-training
Users that are interested in large-batch-training are comparing it to the libraries listed below
Sorting:
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 8 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 7 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- An implementation of shampoo☆74Updated 7 years ago
- Example code for the paper "Understanding deep learning requires rethinking generalization"☆178Updated 5 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Updated 5 years ago
- Lua implementation of Entropy-SGD☆82Updated 7 years ago
- Code for paper "Convergent Learning: Do different neural networks learn the same representations?"☆86Updated 8 years ago
- ☆133Updated 7 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- Implementation of the Deep Frank-Wolfe Algorithm -- Pytorch☆62Updated 4 years ago
- Efficient Architecture Search by Network Transformation, in AAAI 2018☆169Updated 6 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆54Updated 5 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆87Updated 4 years ago
- ☆219Updated 7 years ago
- Neural Architecture Search with Bayesian Optimisation and Optimal Transport☆133Updated 6 years ago
- Net2Net implementation on PyTorch for any possible vision layers.☆38Updated 7 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- PyTorch implementation of PathNet: Evolution Channels Gradient Descent in Super Neural Networks☆80Updated 7 years ago
- A tutorial on "Bayesian Compression for Deep Learning" published at NIPS (2017).☆206Updated 6 years ago
- Videos of deep learning optimizers moving on 3D problem-landscapes☆107Updated 10 months ago
- Release of CIFAR-10.1, a new test set for CIFAR-10.☆222Updated 5 years ago
- A drop-in replacement for CIFAR-10.☆241Updated 4 years ago
- The Singular Values of Convolutional Layers☆72Updated 6 years ago
- Repository of code for the experiments for the ICLR submission "An Empirical Investigation of Catastrophic Forgetting in Gradient-Based N…☆68Updated 11 years ago
- ☆61Updated 2 years ago
- Structured Bayesian Pruning, NIPS 2017☆74Updated 4 years ago
- ☆27Updated 6 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆139Updated 7 years ago
- Minimal Tensorflow implementation of the paper "Neural Architecture Search With Reinforcement Learning" presented at ICLR 2017☆41Updated 7 years ago