yaroslavvb / memory_util
TensorFlow util for building memory usage timeline from LOG_MEMORY messages
☆64Updated 7 years ago
Alternatives and similar repositories for memory_util:
Users that are interested in memory_util are comparing it to the libraries listed below
- Efficient layer normalization GPU kernel for Tensorflow☆111Updated 8 years ago
- ☆23Updated 8 years ago
- A pytorch implementation of "Self-Normalizing Neural Networks" by Klambauer et al. (still beta)☆59Updated 7 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- Reproduction of some of the results from 'Identity Mappings in Deep Residual Networks'☆72Updated 8 years ago
- TensorFlow kernels for probing memory☆15Updated 8 years ago
- ☆28Updated 7 years ago
- Weight initialization schemes for PyTorch nn.Modules☆70Updated 8 years ago
- Python bindings for pyNVML and psutil library over network☆51Updated last year
- Signal Processing Library for PyTorch☆38Updated 7 years ago
- Direct C++ Interface to PyTorch☆80Updated 7 years ago
- Reference caffe implementation of LSUV initialization☆113Updated 7 years ago
- OpenCL Inference Engine for pytorch☆51Updated 7 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 7 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 9 years ago
- ☆36Updated 6 years ago
- Distributed Learning by Pair-Wise Averaging☆52Updated 7 years ago
- ☆69Updated 6 years ago
- An attempt to implement the recurrent attention model (RAM) from "Recurrent Models of Visual Attention" (Mnih+ 2014)☆43Updated 4 years ago
- Example of backprop which uses constant memory☆43Updated 6 years ago
- ☆57Updated 6 years ago
- ☆44Updated 7 years ago
- ☆25Updated 10 years ago
- ☆29Updated 7 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- easy embeddable Torch7 networks☆35Updated 8 years ago
- train on AWS☆75Updated 6 years ago
- Weight initialisation schemes for Torch7 neural network modules☆100Updated 7 years ago
- Code for Attentive Recurrent Comparators☆57Updated 8 years ago
- Lasagne code for weight normalization☆88Updated 9 years ago