A minor typo fix in model maker image classifier hyperparameters python docstring.
PiperOrigin-RevId: 566355937
This commit is contained in:
parent
d5fa4a157e
commit
f4477f1739
|
@ -38,7 +38,7 @@ class HParams(hp.BaseHParams):
|
||||||
decay_samples: Number of training samples used to calculate the decay steps
|
decay_samples: Number of training samples used to calculate the decay steps
|
||||||
and create the training optimizer.
|
and create the training optimizer.
|
||||||
warmup_steps: Number of warmup steps for a linear increasing warmup schedule
|
warmup_steps: Number of warmup steps for a linear increasing warmup schedule
|
||||||
on learning rate. Used to set up warmup schedule by model_util.WarmUp.s
|
on learning rate. Used to set up warmup schedule by model_util.WarmUp.
|
||||||
"""
|
"""
|
||||||
# Parameters from BaseHParams class.
|
# Parameters from BaseHParams class.
|
||||||
learning_rate: float = 0.001
|
learning_rate: float = 0.001
|
||||||
|
|
Loading…
Reference in New Issue
Block a user