Skip to content

gradient-centralization-tf 0.0.1

Choose a tag to compare

@Rishit-dagli Rishit-dagli released this 20 Feb 08:07
· 42 commits to main since this release
2982942

This is the initial version of the Gradient-Centralization-TensorFlow package.

Features:

  • Implement Gradient centralization for optimizers using tf.keras.optimizer.Optimizers base class
  • Supports custom optimizers
  • Pre-built optimizers implementing GC for testing purposes.

Thanks, @ialimustufa for his contributions to this package.