From 1bf289d5f1ac857604d76a6648cc942052b7c985 Mon Sep 17 00:00:00 2001 From: Max Moroz Date: Tue, 26 Oct 2021 20:51:22 -0700 Subject: [PATCH] Minor comment fix to refer to the correct argument name. PiperOrigin-RevId: 405801291 --- mesh_tensorflow/transformer/learning_rate_schedules.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mesh_tensorflow/transformer/learning_rate_schedules.py b/mesh_tensorflow/transformer/learning_rate_schedules.py index 50c6b0a0..c025c085 100644 --- a/mesh_tensorflow/transformer/learning_rate_schedules.py +++ b/mesh_tensorflow/transformer/learning_rate_schedules.py @@ -151,7 +151,8 @@ def constant(step, total_train_steps, value=1.0): def constant_learning_rate(step, total_train_steps, learning_rate=gin.REQUIRED): """Learning rate independent of step. - DEPRECATED: use constant() or pass a float directly to utils.run.learning_rate + DEPRECATED: use constant() or pass a float directly to + utils.run.learning_rate_schedule. Args: step: a tf.Scalar