Commit 7e335e15 authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

added pre ad post process

parent 54093417

tasks/orqa/evaluate_utils.py

+3 −1
Original line number Diff line number Diff line
@@ -47,9 +47,11 @@ class ORQAEvaluator(object): 
        #args.only_query_model = only_query_model

        #args.only_context_model = False



        #model = get_model(lambda: biencoder_model_provider(only_query_model=\

        model = get_model(lambda: biencoder_model_provider(only_query_model=\

            only_query_model, biencoder_shared_query_context_model=\

            args.biencoder_shared_query_context_model))

            args.biencoder_shared_query_context_model,

            pre_process=True, post_process=True))



        #model = get_model(biencoder_model_provider)