Fix lr -> learning_rate

tensorflow · Sep 11, 2024 · f1ba57a · f1ba57a
1 parent 9ba0000
commit f1ba57a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/site/en/tutorials/distribute/keras.ipynb b/site/en/tutorials/distribute/keras.ipynb
@@ -396,7 +396,7 @@
         "# Define a callback for printing the learning rate at the end of each epoch.\n",
         "class PrintLR(tf.keras.callbacks.Callback):\n",
         "  def on_epoch_end(self, epoch, logs=None):\n",
-        "    print('\\nLearning rate for epoch {} is {}'.format(epoch + 1, model.optimizer.lr.numpy()))"
+        "    print('\\nLearning rate for epoch {} is {}'.format(epoch + 1, model.optimizer.learning_rate.numpy()))"
       ]
     },
     {