Skip to content

v0.4.0

Compare
Choose a tag to compare
@andreped andreped released this 15 Apr 10:59
· 195 commits to main since this release
8179c8a

What's Changed

  • Added custom AccumBatchNormalization layer with gradient accumulation support.
  • Added more unit tests -> code coverage = 99%
  • Made proper documentations which is hosted at gradientaccumulator.readthedocs.io/
  • Reduced runtime on several unit tests to make CI jobs faster
  • Added fix for protobuf for tfds in CIs
  • Reworked README - moved most stuff to the Documentations + added CI section w/badges
  • Header image by @jpdefrutos in #51

New Contributors

New API feature

from gradient_accumulator import AccumBatchNormalization

layer = AccumBatchNormalization(accum_steps=4)

Can be used as a regular Keras BatchNormalization layer, but with reduced functionality.

Full Changelog: v0.3.2...v0.4.0