[LIGHT] Who Am I - Preventing doubleloading model (facebookresearch#4742

)
CZWin32768 · Aug 15, 2022 · c17f8ed · c17f8ed
1 parent 59ec67c
commit c17f8ed
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/projects/light_whoami/agents/pacer.py b/projects/light_whoami/agents/pacer.py
@@ -74,10 +74,14 @@ def add_cmdline_args(
     def __init__(self, opt: Opt, shared=None):
         super().__init__(opt, shared)
         reranker_class = self.get_partial_only_reranker_class()
-        if not (shared and 'classifier' in shared):
-            self.classifier = reranker_class(opt)
-        else:
+        if shared and 'classifier' in shared:
             self.classifier = shared['classifier']
+        elif shared and 'reranker' in shared:
+            self.classifier = shared['reranker']
+        elif hasattr(self, 'reranker'):
+            self.classifier = self.reranker
+        else:
+            self.classifier = reranker_class(opt)
         assert opt[
             'beam_block_full_context'
         ], 'must set --beam-block-full-context True to use PACER'