We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2e69d1b commit 3325b71Copy full SHA for 3325b71
test/smoke_test.py
@@ -59,7 +59,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
59
model.eval()
60
61
# Step 2: Initialize the inference transforms
62
- preprocess = weights.transforms()
+ preprocess = weights.transforms(antialias=(device != "mps")) # antialias not supported on MPS
63
64
# Step 3: Apply inference preprocessing transforms
65
batch = preprocess(img).unsqueeze(0)
0 commit comments