`tfa.optimizers.MovingAverage` & `tfa.optimizers.SWA` future

tensorflow / addons

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons

Apache License 2.0

1.69k stars 610 forks source link

`tfa.optimizers.MovingAverage` & `tfa.optimizers.SWA` future #2844

Closed aaarrti closed 1 year ago

aaarrti commented 1 year ago

Hi,

The average optimizer wrappers support only tf.keras.optimizers.legacy optimizers. Since TFA was deprecated, I doubt there will be not development work here to support the new APIs. Are there any plans to support average optimizers in https://github.com/keras-team/keras-core ? Is there any replacement compatible with newer Keras optimizers?

bhack commented 1 year ago

I suggest to open this ticket in Keras or Keras-core repo as feature request.