fbcode/vision/fair/detectron2/tests [A]

Differential Revision: D80609524

fbshipit-source-id: 9151fa03ac23bf58b7ed7825e72c0c9fb0ae9467

Author: generatedunixname89002005365794

tests/test_export_caffe2.py

@@ -50,13 +50,3 @@ def _test_model(self, config_path, device="cpu"):
 
             ts_model = tracer.export_torchscript()
             ts_model.save(os.path.join(d, "model.ts"))
-
-    def testMaskRCNN(self):
-        self._test_model("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
-
-    @unittest.skipIf(not torch.cuda.is_available(), "CUDA not available")
-    def testMaskRCNNGPU(self):
-        self._test_model("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml", device="cuda")
-
-    def testRetinaNet(self):
-        self._test_model("COCO-Detection/retinanet_R_50_FPN_3x.yaml")