cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

A minor typo fix in model maker image classifier hyperparameters python docstring.

Browse Source 
PiperOrigin-RevId: 566355937
This commit is contained in:
MediaPipe Team 2023-09-18 11:06:04 -07:00 committed by Copybara-Service
parent d5fa4a157e
commit f4477f1739
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/model_maker/python/vision/image_classifier/hyperparameters.py
View File

@ -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