keskarnitish / large-batch-training
Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"
☆138Updated 7 years ago
Related projects ⓘ
Alternatives and complementary repositories for large-batch-training
- 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
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 6 years ago
- Example code for the paper "Understanding deep learning requires rethinking generalization"☆177Updated 4 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- A drop-in replacement for CIFAR-10.☆236Updated 3 years ago
- ☆132Updated 7 years ago
- Lua implementation of Entropy-SGD☆81Updated 6 years ago
- Principled Detection of Out-of-Distribution Examples in Neural Networks☆200Updated 7 years ago
- An implementation of shampoo☆74Updated 6 years ago
- This repository is no longer maintained. Check☆82Updated 4 years ago
- Code for paper "Convergent Learning: Do different neural networks learn the same representations?"☆85Updated 8 years ago
- ☆44Updated 6 years ago
- Release of CIFAR-10.1, a new test set for CIFAR-10.☆220Updated 4 years ago
- Snapshot Ensembles in Torch (Snapshot Ensembles: Train 1, Get M for Free)☆190Updated 7 years ago
- ☆219Updated 6 years ago
- Repository of code for the experiments for the ICLR submission "An Empirical Investigation of Catastrophic Forgetting in Gradient-Based N…☆66Updated 10 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆135Updated 5 years ago
- A tutorial on "Bayesian Compression for Deep Learning" published at NIPS (2017).☆205Updated 5 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- Implementation of the Deep Frank-Wolfe Algorithm -- Pytorch☆61Updated 3 years ago
- Visualizing Deep Neural Network Decisions: Prediction Difference Analysis☆117Updated 7 years ago
- Implementation of soft parameter sharing for neural networks☆69Updated 3 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 4 years ago
- ☆250Updated 7 years ago
- ☆26Updated 5 years ago
- Structured Bayesian Pruning, NIPS 2017☆73Updated 4 years ago
- Basic experiment framework for tensorflow.☆92Updated 3 years ago
- ☆64Updated 8 years ago
- Hypergradient descent☆138Updated 5 months ago