fix context in multitask (#1040)

Summary: Pull Request resolved: #1040 Context is not handled propertly in multitask. This should fix it. Differential Revision: D17860982 fbshipit-source-id: 4121ce6bd6d57e3fdfcb757756010ede30e92270
facebookresearch · Oct 11, 2019 · 0bdadb4 · 0bdadb4
1 parent fafe613
commit 0bdadb4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytext/models/disjoint_multitask_model.py b/pytext/models/disjoint_multitask_model.py
@@ -64,5 +64,8 @@ def arrange_model_inputs(self, tensor_dict):
     def arrange_targets(self, tensor_dict):
         return self.current_model.arrange_targets(tensor_dict)
 
+    def arrange_model_context(self, tensor_dict):
+        return self.current_model.arrange_model_context(tensor_dict)
+
     def caffe2_export(self, tensorizers, tensor_dict, path, export_onnx_path=None):
         pass