626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

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

Browse source 
DP optimizer only supports legacy optimizer.

PiperOrigin-RevId: 474137890
This commit is contained in:
Chen Qian 2022-09-13 15:19:44 -07:00 committed by A. Unique TensorFlower
parent 08364adcb7
commit c6c3334b57
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tensorflow_privacy/privacy/optimizers/dp_optimizer_keras_vectorized.py
View file

@ -252,8 +252,8 @@ def make_vectorized_keras_optimizer_class(cls):
VectorizedDPKerasAdagradOptimizer = make_vectorized_keras_optimizer_class(
tf.keras.optimizers.Adagrad)
tf.keras.optimizers.legacy.Adagrad)
VectorizedDPKerasAdamOptimizer = make_vectorized_keras_optimizer_class(
tf.keras.optimizers.Adam)
tf.keras.optimizers.legacy.Adam)
VectorizedDPKerasSGDOptimizer = make_vectorized_keras_optimizer_class(
tf.keras.optimizers.SGD)
tf.keras.optimizers.legacy.SGD)