Skip to content

Commit 4f0745b

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] disable download progress for smoke test weights (#8158)
Reviewed By: vmoens Differential Revision: D52539018 fbshipit-source-id: 2659afc0aa0b0094d84ec4bb9b0d4f509894815b
1 parent faab042 commit 4f0745b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/smoke_test.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
5353

5454
# Step 1: Initialize model with the best available weights
5555
weights = ResNet50_Weights.DEFAULT
56-
model = resnet50(weights=weights).to(device)
56+
model = resnet50(weights=weights, progress=False).to(device)
5757
model.eval()
5858

5959
# Step 2: Initialize the inference transforms

0 commit comments

Comments
 (0)