delete caffe2 export (#1307)

Summary: Pull Request resolved: #1307 delete caffe2 export 1. BiTransformer 2. NewBertModel 3. NewDisjointMultitaskModel Reviewed By: snisarg Differential Revision: D20907342 fbshipit-source-id: f8579dc68d95c92549af4e79d67723e6915d31b5
facebookresearch · Apr 9, 2020 · 6abc1e7 · 6abc1e7
1 parent 9548c94
commit 6abc1e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytext/models/bert_classification_models.py b/pytext/models/bert_classification_models.py
@@ -83,7 +83,7 @@ def forward(
         return self.decoder(representation, *args)
 
     def caffe2_export(self, tensorizers, tensor_dict, path, export_onnx_path=None):
-        pass
+        raise NotImplementedError
 
     @classmethod
     def from_config(cls, config: Config, tensorizers: Dict[str, Tensorizer]):

diff --git a/pytext/models/disjoint_multitask_model.py b/pytext/models/disjoint_multitask_model.py
@@ -70,4 +70,4 @@ 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
+        raise NotImplementedError