megatron.model.biencoder_model

Description

Classes

BiEncoderModel([num_tokentypes, ...])	Bert-based module for Biencoder model.
PretrainedBertModel([num_tokentypes, ...])	BERT-based encoder for queries or contexts used for learned information retrieval.

Functions

biencoder_model_provider([only_query_model, ...])	Build the model.
get_model_provider([only_query_model, ...])