Module: mediapipe_model_maker.text_classifier.model_spec.bert_model_spec

Specification for a BERT model.

View aliases

Main aliases

mediapipe_model_maker.python.text.core.bert_model_spec, mediapipe_model_maker.python.text.text_classifier.model_spec.bert_model_spec, mediapipe_model_maker.python.text.text_classifier.text_classifier.ms.bert_model_spec, mediapipe_model_maker.python.text.text_classifier.text_classifier.text_classifier_options.ms.bert_model_spec, mediapipe_model_maker.python.text.text_classifier.text_classifier_options.ms.bert_model_spec, mediapipe_model_maker.text_classifier.text_classifier.ms.bert_model_spec, mediapipe_model_maker.text_classifier.text_classifier.text_classifier_options.ms.bert_model_spec, mediapipe_model_maker.text_classifier.text_classifier_options.ms.bert_model_spec

Modules

bert_model_options module: Configurable model options for a BERT model.

hp module: Hyperparameters for training models. Shared across tasks.

Classes

class BertModelSpec: Specification for a BERT model.