Wrapping optimizer for gradient accumulation in TF2 and Keras2
It supports mixed precision and multi-GPUs and tested with two GTX 1080Ti in TF 2.16.1 and Keras 2.16
-
Notifications
You must be signed in to change notification settings - Fork 0
Wrapping optimizer for gradient accumulation in TF2 and Keras2
License
pinesnow72/gaoptimizer
About
Wrapping optimizer for gradient accumulation in TF2 and Keras2
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published