Adabelief-Optimizer
Adabelief-Optimizer copied to clipboard
adapt to Tensorflow >= 2.11
As of TF 2.11, Optimizer must be imported from tf.keras.optimizers.legacy
instead of tf.keras.optimizers
.
I have tested with TF 2.13.1 on AdaBelief_tf2_test.py from the 0.3.0 branch (with a slight modification of the assertion to ignore extra config details).
If you believe I should rebase on the 0.3.0 branch, please let me know. (That changeset looks broken to me – it includes lots of compilation and cache results.)