diff --git a/python/ray/tune/examples/pbt_tune_cifar10_with_keras.py b/python/ray/tune/examples/pbt_tune_cifar10_with_keras.py index 11163f722..28575f546 100755 --- a/python/ray/tune/examples/pbt_tune_cifar10_with_keras.py +++ b/python/ray/tune/examples/pbt_tune_cifar10_with_keras.py @@ -110,7 +110,8 @@ class Cifar10Model(Trainable): x_train = self.train_data[0] model = self._build_model(x_train.shape[1:]) - opt = tf.keras.optimizers.Adadelta() + opt = tf.keras.optimizers.Adadelta( + lr=self.config["lr"], decay=self.config["decay"]) model.compile( loss="categorical_crossentropy", optimizer=opt,