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 8ea4772 commit 7b7f0b0Copy full SHA for 7b7f0b0
test/smoke_test.py
@@ -96,7 +96,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
96
model.eval()
97
98
# Step 2: Initialize the inference transforms
99
- preprocess = weights.transforms(antialias=(device != "mps")) # antialias not supported on MPS
+ preprocess = weights.transforms(antialias=True)
100
101
# Step 3: Apply inference preprocessing transforms
102
batch = preprocess(img).unsqueeze(0)
0 commit comments