gradient-accumulation topic
List
gradient-accumulation repositories
koila
1.8k
Stars
62
Forks
Watchers
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
ssd_detectors
297
Stars
79
Forks
Watchers
SSD-based object and text detection with Keras, SSD, DSOD, TextBoxes, SegLink, TextBoxes++, CRNN
keras-gradient-accumulation
35
Stars
4
Forks
Watchers
Gradient accumulation for Keras
GradientAccumulator
50
Stars
10
Forks
Watchers
:dart: Accumulated Gradients for TensorFlow 2
torch-handle
19
Stars
5
Forks
Watchers
TorchHandle makes your PyTorch development more efficient and make you use PyTorch more comfortable
pytorch-transformer-distributed
24
Stars
10
Forks
Watchers
Distributed training (multi-node) of a Transformer model