Skip to content

GitLab

Unverified Commit 7e9e7844 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

disable download progress for smoke test weights (#8158)

parent 490929c5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/smoke_test.py
View file @ 7e9e7844
...@@ -53,7 +53,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None: ...@@ -53,7 +53,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
# Step 1: Initialize model with the best available weights # Step 1: Initialize model with the best available weights
weights = ResNet50_Weights.DEFAULT weights = ResNet50_Weights.DEFAULT
model = resnet50(weights=weights).to(device) model = resnet50(weights=weights, progress=False).to(device)
model.eval() model.eval()
# Step 2: Initialize the inference transforms # Step 2: Initialize the inference transforms
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment