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

Internal model maker change

Browse Source 
PiperOrigin-RevId: 497196512
This commit is contained in:
MediaPipe Team 2022-12-22 10:41:03 -08:00 committed by Copybara-Service
parent 5b90afda70
commit 36f054dfbe
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mediapipe/model_maker/python/text/text_classifier/text_classifier.py
View File

@ -33,7 +33,6 @@ from mediapipe.model_maker.python.text.text_classifier import preprocessor
from mediapipe.model_maker.python.text.text_classifier import text_classifier_options from mediapipe.model_maker.python.text.text_classifier import text_classifier_options
from mediapipe.tasks.python.metadata.metadata_writers import metadata_writer from mediapipe.tasks.python.metadata.metadata_writers import metadata_writer
from mediapipe.tasks.python.metadata.metadata_writers import text_classifier as text_classifier_writer from mediapipe.tasks.python.metadata.metadata_writers import text_classifier as text_classifier_writer
from official.nlp import optimization
def _validate(options: text_classifier_options.TextClassifierOptions): def _validate(options: text_classifier_options.TextClassifierOptions):
@ -424,7 +423,7 @@ class _BertClassifier(TextClassifier):
end_learning_rate=0.0, end_learning_rate=0.0,
power=1.0) power=1.0)
if warmup_steps: if warmup_steps:
lr_schedule = optimization.WarmUp( lr_schedule = model_util.WarmUp(
initial_learning_rate=initial_lr, initial_learning_rate=initial_lr,
decay_schedule_fn=lr_schedule, decay_schedule_fn=lr_schedule,
warmup_steps=warmup_steps) warmup_steps=warmup_steps)