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
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file keras-gradient-accumulation-0.0.6.tar.gz.
File metadata
- Download URL: keras-gradient-accumulation-0.0.6.tar.gz
- Upload date:
- Size: 7.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/1.13.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.2 CPython/3.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1e5046aeafdcedb88e5d047b2f2791366a8574a24de5a1f79b378873e53db7b2
|
|
| MD5 |
23a55ae1e265d52c675750b91a9ba832
|
|
| BLAKE2b-256 |
bcaddac21e613c0016f6b145d434faaa3a0715f8bb40769d35ba1c2f206597de
|
