Skip to content

Commit 8b7275b

Browse files
committed
Use torchvision directly
Was getting throttling errors when trying to access torch hub from GitHub actions.
1 parent 7fd560d commit 8b7275b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/e2e/test_torch_serialize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
# --------------------------------------------------------------------------
66
import pytest
77
import torch
8+
import torchvision.models
89

910
from azstoragetorch.io import BlobIO
1011
from utils import random_resource_name
@@ -20,8 +21,7 @@ def torch_hub_cache(tmp_path_factory):
2021

2122
@pytest.fixture(scope="module")
2223
def model():
23-
model = torch.hub.load("pytorch/vision:v0.10.0", "resnet101", pretrained=False)
24-
return model
24+
return torchvision.models.resnet101()
2525

2626

2727
@pytest.fixture(scope="module")

0 commit comments

Comments
 (0)