disable download progress for smoke test weights (#8158)

pytorch · Dec 11, 2023 · 7e9e784 · 7e9e784
1 parent 490929c
commit 7e9e784
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/smoke_test.py b/test/smoke_test.py
@@ -53,7 +53,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
 
     # Step 1: Initialize model with the best available weights
     weights = ResNet50_Weights.DEFAULT
-    model = resnet50(weights=weights).to(device)
+    model = resnet50(weights=weights, progress=False).to(device)
     model.eval()
 
     # Step 2: Initialize the inference transforms