File tree 1 file changed +20
-0
lines changed
torchvision/prototype/models/quantization 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -87,6 +87,16 @@ class QuantizedResNet50Weights(Weights):
87
87
"acc@5" : 92.814 ,
88
88
},
89
89
)
90
+ ImageNet1K_FBGEMM_RefV2 = WeightEntry (
91
+ url = "https://download.pytorch.org/models/quantized/resnet50_fbgemm-23753f79.pth" ,
92
+ transforms = partial (ImageNetEval , crop_size = 224 , resize_size = 232 ),
93
+ meta = {
94
+ ** _COMMON_META ,
95
+ "unquantized" : ResNet50Weights .ImageNet1K_RefV2 ,
96
+ "acc@1" : 80.282 ,
97
+ "acc@5" : 94.976 ,
98
+ },
99
+ )
90
100
91
101
92
102
class QuantizedResNeXt101_32x8dWeights (Weights ):
@@ -100,6 +110,16 @@ class QuantizedResNeXt101_32x8dWeights(Weights):
100
110
"acc@5" : 94.480 ,
101
111
},
102
112
)
113
+ ImageNet1K_FBGEMM_RefV2 = WeightEntry (
114
+ url = "https://download.pytorch.org/models/quantized/resnext101_32x8_fbgemm-ee16d00c.pth" ,
115
+ transforms = partial (ImageNetEval , crop_size = 224 , resize_size = 232 ),
116
+ meta = {
117
+ ** _COMMON_META ,
118
+ "unquantized" : ResNeXt101_32x8dWeights .ImageNet1K_RefV2 ,
119
+ "acc@1" : 82.574 ,
120
+ "acc@5" : 96.132 ,
121
+ },
122
+ )
103
123
104
124
105
125
def resnet18 (
You can’t perform that action at this time.
0 commit comments