Skip to content

Commit aa1adfb

Browse files
committed
Correct meta info.
1 parent 0ce4670 commit aa1adfb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torchvision/prototype/models/quantization/inception.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,17 +25,17 @@
2525
class QuantizedInceptionV3Weights(Weights):
2626
ImageNet1K_FBGEMM_TFV1 = WeightEntry(
2727
url="https://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-71447a44.pth",
28-
transforms=partial(ImageNetEval, crop_size=224),
28+
transforms=partial(ImageNetEval, crop_size=299, resize_size=342),
2929
meta={
30-
"size": (224, 224),
30+
"size": (299, 299),
3131
"categories": _IMAGENET_CATEGORIES,
3232
"interpolation": InterpolationMode.BILINEAR,
3333
"backend": "fbgemm",
3434
"quantization": "ptq",
3535
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-models",
3636
"unquantized": InceptionV3Weights.ImageNet1K_TFV1,
37-
"acc@1": 69.826,
38-
"acc@5": 89.404,
37+
"acc@1": 77.176,
38+
"acc@5": 93.354,
3939
},
4040
)
4141

0 commit comments

Comments
 (0)