updated the evaluation script for retriver

f21a6628 · Mostofa Patwary · dfb6a9b3 · f21a6628
Commit f21a6628 authored May 18, 2021 by Mostofa Patwary
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

megatron/data/biencoder_dataset_utils.py megatron/data/biencoder_dataset_utils.py +0 -1

No files found.
--- a/megatron/data/biencoder_dataset_utils.py
+++ b/megatron/data/biencoder_dataset_utils.py
@@ -20,7 +20,6 @@ def make_attention_mask(source_block, target_block):
    # (source_length, target_length)
    return mask

-
 def get_one_epoch_dataloader(dataset, micro_batch_size=None):
    """Specifically one epoch to be used in an indexing job."""
    args = get_args()