Internal change

PiperOrigin-RevId: 341074795

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
     })