updated the evaluation script for retriver

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

Showing with 1 addition and 1 deletion

megatron/model/biencoder_model.py megatron/model/biencoder_model.py +1 -1

No files found.
--- a/megatron/model/biencoder_model.py
+++ b/megatron/model/biencoder_model.py
@@ -26,7 +26,7 @@ def get_model_provider(only_query_model=False, only_context_model=False,
                only_context_model = only_context_model,
                biencoder_shared_query_context_model = \
                biencoder_shared_query_context_model,
-                pre_process=True, post_process=True)
+                pre_process=pre_process, post_process=post_process)

        return model