added pre ad post process

7e335e15 · Mostofa Patwary · 54093417 · 7e335e15
Commit 7e335e15 authored May 12, 2021 by Mostofa Patwary
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

tasks/orqa/evaluate_utils.py tasks/orqa/evaluate_utils.py +3 -1

No files found.
--- a/tasks/orqa/evaluate_utils.py
+++ b/tasks/orqa/evaluate_utils.py
@@ -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)