Internal change

PiperOrigin-RevId: 341074795

Internal change
PiperOrigin-RevId: 341074795
63665121 · Le Hou · A. Unique TensorFlower · eef4e2dd · 63665121
Commit 63665121 authored Nov 06, 2020 by Le Hou Committed by A. Unique TensorFlower Nov 06, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/modeling/optimization/lr_schedule.py official/modeling/optimization/lr_schedule.py +1 -1

No files found.
--- a/official/modeling/optimization/lr_schedule.py
+++ b/official/modeling/optimization/lr_schedule.py
@@ -148,7 +148,7 @@ class PolynomialWarmUp(tf.keras.optimizers.schedules.LearningRateSchedule):
      config = {"after_warmup_lr_sched": self._after_warmup_lr_sched}  # pytype: disable=attribute-error

    config.update({
-        "warmup_steps": self._warmup_setps,
+        "warmup_steps": self._warmup_steps,
        "power": self._power,
        "name": self._name
    })