[cleanup] Remove disabled e2e test

This test has been disabled a long time, and since RefBackend is so slow
we don't want to add this unnecessarily. I believe it is covered by
downstream testing such as the Shark Tank.
pull/1696/head
Sean Silva 2022-12-07 14:00:30 +00:00
parent d52359a891
commit c956c39c86
1 changed files with 0 additions and 22 deletions

View File

@ -84,28 +84,6 @@ class IouOfModule(torch.nn.Module):
def IouOfModule_basic(module, tu: TestUtils):
module.forward(tu.rand(1024, 4), tu.rand(1024, 4))
class MobilenetV2Module(torch.nn.Module):
def __init__(self):
super().__init__()
# Reset seed to make model deterministic.
torch.manual_seed(0)
self.mobilenetv2 = models.mobilenet_v2()
self.train(False)
@export
@annotate_args([
None,
([-1, 3, -1, -1], torch.float32, True),
])
def forward(self, img):
return self.mobilenetv2.forward(img)
# TODO (cathyzhyi) The runtime assertion for conv2d with group != 1 is exposed
# after aten.hardtanh is implemented. Reenable once the the runtime assertion
# is fixed.
#@register_test_case(module_factory=lambda: MobilenetV2Module())
def MobilenetV2Module_basic(module, tu: TestUtils):
module.forward(tu.rand(1, 3, 224, 224))
class MobilenetV3Module(torch.nn.Module):
def __init__(self):