Commit a41e4781 authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

updated the evaluation script for retriver

parent f21a6628

megatron/model/biencoder_model.py

+1 −1
Original line number Diff line number Diff line
@@ -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