We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 490929c commit 7e9e784Copy full SHA for 7e9e784
test/smoke_test.py
@@ -53,7 +53,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
53
54
# Step 1: Initialize model with the best available weights
55
weights = ResNet50_Weights.DEFAULT
56
- model = resnet50(weights=weights).to(device)
+ model = resnet50(weights=weights, progress=False).to(device)
57
model.eval()
58
59
# Step 2: Initialize the inference transforms
0 commit comments