From 87bca0262b284a04f5ffb269ed51df5942afeaed Mon Sep 17 00:00:00 2001 From: Barlas Oguz Date: Thu, 10 Oct 2019 12:22:10 -0700 Subject: [PATCH] fix context in multitask Summary: Context is not handled propertly in multitask. This should fix it. Differential Revision: D17860982 fbshipit-source-id: 398015b4d62ec00db20f789385dfc9c436192e29 --- pytext/models/disjoint_multitask_model.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pytext/models/disjoint_multitask_model.py b/pytext/models/disjoint_multitask_model.py index 41c3b9c41..5f643318d 100644 --- 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