rdspring1 / Count-Sketch-Optimizers
A compressed adaptive optimizer for training large-scale deep learning models using PyTorch
☆27Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for Count-Sketch-Optimizers
- ☆73Updated 5 years ago
- Sketched SGD☆28Updated 4 years ago
- ☆43Updated 4 years ago
- Sparsified SGD with Memory: https://arxiv.org/abs/1809.07599☆55Updated 6 years ago
- Simple Hierarchical Count Sketch in Python☆20Updated 3 years ago
- CoLa - Decentralized Linear Learning: https://arxiv.org/abs/1808.04883☆19Updated 2 years ago
- Code for the signSGD paper☆80Updated 3 years ago
- Implementation of (overlap) local SGD in Pytorch☆32Updated 4 years ago
- ☆29Updated 4 years ago
- implement distributed machine learning with Pytorch + OpenMPI☆51Updated 5 years ago
- SmoothOut: Smoothing Out Sharp Minima to Improve Generalization in Deep Learning☆23Updated 5 years ago
- Code for "On the Relation Between the Sharpest Directions of DNN Loss and the SGD Step Length", ICLR 2019☆11Updated last year
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- A PyTorch implementation of the paper "Decoupled Parallel Backpropagation with Convergence Guarantee"☆30Updated 6 years ago
- SGD with compressed gradients and error-feedback: https://arxiv.org/abs/1901.09847☆29Updated 3 months ago
- Stochastic Gradient Push for Distributed Deep Learning☆157Updated last year
- Practical low-rank gradient compression for distributed optimization: https://arxiv.org/abs/1905.13727☆143Updated last week
- Code for ICML 2017 paper, SplitNet: Learning to Semantically Split Deep Networks for Parameter Reduction and Model Parallelization☆55Updated 7 years ago
- Atomo: Communication-efficient Learning via Atomic Sparsification☆25Updated 5 years ago
- ☆27Updated last year
- ☆82Updated 4 years ago
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆100Updated 4 years ago
- QSGD-TF☆21Updated 5 years ago
- Reproduction and analysis of SNIP paper☆28Updated 4 years ago
- This repo contains the code used for NeurIPS 2019 paper "Asymmetric Valleys: Beyond Sharp and Flat Local Minima".☆14Updated 5 years ago
- Net2Net implementation on PyTorch for any possible vision layers.☆38Updated 7 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆46Updated 3 years ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆20Updated 4 years ago
- Compressing Neural Networks using the Variational Information Bottleneck☆64Updated 2 years ago