Gradient accumulation for Keras
Project description
Keras Gradient Accumulation
Install
pip install keras-gradient-accumulation
Usage
Wrapper
from keras_gradient_accumulation import GradientAccumulation
optimizer = GradientAccumulation('adam', accumulation_steps=8)
Adam
from keras_gradient_accumulation import AdamAccumulated
optimizer = AdamAccumulated(accumulation_steps=8)
Known Issues
- Not available for batch normalization
- Not compatible with
OptimizerV2
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Close
Hashes for keras-gradient-accumulation-0.0.6.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 1e5046aeafdcedb88e5d047b2f2791366a8574a24de5a1f79b378873e53db7b2 |
|
| MD5 | 23a55ae1e265d52c675750b91a9ba832 |
|
| BLAKE2b-256 | bcaddac21e613c0016f6b145d434faaa3a0715f8bb40769d35ba1c2f206597de |
