File tree 1 file changed +4
-4
lines changed
torchvision/prototype/models/quantization
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 25
25
class QuantizedInceptionV3Weights (Weights ):
26
26
ImageNet1K_FBGEMM_TFV1 = WeightEntry (
27
27
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 ),
29
29
meta = {
30
- "size" : (224 , 224 ),
30
+ "size" : (299 , 299 ),
31
31
"categories" : _IMAGENET_CATEGORIES ,
32
32
"interpolation" : InterpolationMode .BILINEAR ,
33
33
"backend" : "fbgemm" ,
34
34
"quantization" : "ptq" ,
35
35
"recipe" : "https://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-models" ,
36
36
"unquantized" : InceptionV3Weights .ImageNet1K_TFV1 ,
37
- "acc@1" : 69.826 ,
38
- "acc@5" : 89.404 ,
37
+ "acc@1" : 77.176 ,
38
+ "acc@5" : 93.354 ,
39
39
},
40
40
)
41
41
You can’t perform that action at this time.
0 commit comments