Rishit-dagli / Gradient-Centralization-TensorFlow

Instantly improve your training performance of TensorFlow models with just 2 lines of code!

Home Page:https://pypi.org/project/gradient-centralization-tf/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Custom Optimizer Example

Rishit-dagli opened this issue · comments

It would be great to include an example showing the use of this package with a custom optimizer built on top of tf.keras.Optimizer class.

On another note I made an example demonstrating this but I soon realized it did not hold anything to offer which the existing two examples do not already (though the existing examples are quite similar themselves I thought it would be better to have at the minimum 2 examples). The code to use gctf is almost the same with a custom optimizer, further, the README also mentions the process for custom optimizers.