Skip to content

Commit 9214973

Browse files
committed
Speed up quantized model tests.
1 parent 1aca7ba commit 9214973

3 files changed

+1
-1
lines changed
Binary file not shown.
Binary file not shown.

test/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -803,7 +803,7 @@ def test_quantized_classification_model(model_fn):
803803
set_rng_seed(0)
804804
defaults = {
805805
"num_classes": 5,
806-
"input_shape": (1, 3, 224, 224),
806+
"input_shape": (1, 3, 64, 64),
807807
"pretrained": False,
808808
"quantize": True,
809809
}

0 commit comments

Comments
 (0)