google / uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.
Apache License 2.0
1.46k stars 203 forks source link

Code changes to get ready for an incoming Keras optimizer migration. #1153

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

Code changes to get ready for an incoming Keras optimizer migration.

Because this code is depending on TFA optimizer, we need to use legacy Keras optimizer.