Skip to content

Commit c6c3334

Browse files
Code changes to get ready for an incoming Keras optimizer migration.
DP optimizer only supports legacy optimizer. PiperOrigin-RevId: 474137890
1 parent 08364ad commit c6c3334

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorflow_privacy/privacy/optimizers/dp_optimizer_keras_vectorized.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -252,8 +252,8 @@ def apply_gradients(self, *args, **kwargs):
252252

253253

254254
VectorizedDPKerasAdagradOptimizer = make_vectorized_keras_optimizer_class(
255-
tf.keras.optimizers.Adagrad)
255+
tf.keras.optimizers.legacy.Adagrad)
256256
VectorizedDPKerasAdamOptimizer = make_vectorized_keras_optimizer_class(
257-
tf.keras.optimizers.Adam)
257+
tf.keras.optimizers.legacy.Adam)
258258
VectorizedDPKerasSGDOptimizer = make_vectorized_keras_optimizer_class(
259-
tf.keras.optimizers.SGD)
259+
tf.keras.optimizers.legacy.SGD)

0 commit comments

Comments
 (0)