Optimizer don't have apply() method

[konstantin-frolov]

Environment:

Python 3.12.7
Tensorflow 2.16.1
Keras 3.8.0

Problem:

All optimizers from tf.keras.optimizers do not have a method apply() to write the train routine from scratch.
But the docs states that it exists.

AttributeError                            Traceback (most recent call last)
Cell In[11], line 1
----> 1 getattr(tf.keras.optimizers.Optimizer(name="test"), "apply")

AttributeError: 'Optimizer' object has no attribute 'apply'

Possible solution

Need to use method apply_gradients(zip(gradients, model_parameters))

[dhantule]

@konstantin-frolov, thanks for reporting this.

If you're trying to apply gradients to variable you can use apply_gradients(zip(gradients, model_parameters)), you can read more about theapply_gradients method here. If that doesn't work please provide some minimal reproducible code. Attaching gist for reference.