[fx] fixed unit tests for torch 1.12 (#1327)

2025-09-09 04:50:17 +00:00 · 2022-07-15 18:22:15 +08:00
parent d49708ae43
commit b2475d8c5c
17 changed files with 2 additions and 18 deletions
--- a/tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py
+++ b/tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py
@@ -19,7 +19,6 @@ torch.manual_seed(MANUAL_SEED)
 torch.backends.cudnn.deterministic = True


-@pytest.mark.skip('skip as torchvision is required')
 def test_torchvision_models():
    MODEL_LIST = [
        tm.vgg11, tm.resnet18, tm.densenet121, tm.mobilenet_v3_small, tm.resnext50_32x4d, tm.wide_resnet50_2,