Skip to content

GitLab

Commit 63665121 authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 341074795
parent eef4e2dd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/modeling/optimization/lr_schedule.py
View file @ 63665121
...@@ -148,7 +148,7 @@ class PolynomialWarmUp(tf.keras.optimizers.schedules.LearningRateSchedule): ...@@ -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 = {"after_warmup_lr_sched": self._after_warmup_lr_sched} # pytype: disable=attribute-error
config.update({ config.update({
"warmup_steps": self._warmup_setps, "warmup_steps": self._warmup_steps,
"power": self._power, "power": self._power,
"name": self._name "name": self._name
}) })
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment