megatron.data.orqa_wiki_dataset.build_sample#

megatron.data.orqa_wiki_dataset.build_sample(row_id, context_ids, context_types, context_pad_mask)#

Convert to numpy and return a sample consumed by the batch producer.