yaroslavvb / chain_constant_memoryLinks
Example of backprop which uses constant memory
☆43Updated 7 years ago
Alternatives and similar repositories for chain_constant_memory
Users that are interested in chain_constant_memory are comparing it to the libraries listed below
Sorting:
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 8 years ago
- 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
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 6 years ago
- Use TensorFlow efficiently☆96Updated 4 years ago
- Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"☆145Updated 8 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 7 years ago
- DAWNBench: An End-to-End Deep Learning Benchmark and Competition☆263Updated 5 years ago
- Files to create the figures in the paper "Super-Convergence: Very Fast Training of Residual Networks Using Large Learning Rates"☆191Updated 8 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- ☆37Updated 6 years ago
- 📈 TensorFlow + Matplotlib as TF ops☆299Updated 6 years ago
- Examples of C extensions for PyTorch☆256Updated 2 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- Prunable nn layers for pytorch.☆48Updated 7 years ago
- train on AWS☆75Updated 7 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 7 years ago
- ☆54Updated 7 years ago
- Direct C++ Interface to PyTorch☆81Updated 7 years ago
- a lightweight and simple logger for Machine Learning☆127Updated 5 years ago
- Stuff I uploaded to share online or to access from a different machine☆174Updated 3 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Updated 5 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆44Updated 8 years ago
- A new kind of pooling layer for faster and sharper convergence☆76Updated 8 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated last year
- CuPy fused PyTorch neural networks ops☆273Updated 7 years ago
- implement distributed machine learning with Pytorch + OpenMPI☆53Updated 6 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆111Updated 3 years ago
- Lasagne code for weight normalization☆88Updated 9 years ago
- repo that holds code for improving on dropout using Stochastic Delta Rule☆141Updated 6 years ago