heterogeneous-distributed-training-framework/docs/source/api-guide/models.bert.rst

models.bert package

Useful package for training bert and bert like encoder only models. It optionally comes with a binary head that can be used for classification tasks .

Submodules

models.bert.bert_model module

core.models.bert.bert_model

Module contents

core.models.bert