Skip to content

Commit 3003dff

Browse files
prabhat00155facebook-github-bot
authored andcommitted
[fbsync] Adding new Quantized models. (#4969)
Reviewed By: NicolasHug Differential Revision: D32694325 fbshipit-source-id: ee2e725d41ee9256a97f86bfe4bd39e5012f7648
1 parent e954ca7 commit 3003dff

File tree

1 file changed

+20
-0
lines changed
  • torchvision/prototype/models/quantization

1 file changed

+20
-0
lines changed

torchvision/prototype/models/quantization/resnet.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,16 @@ class QuantizedResNet50Weights(Weights):
8787
"acc@5": 92.814,
8888
},
8989
)
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+
)
90100

91101

92102
class QuantizedResNeXt101_32x8dWeights(Weights):
@@ -100,6 +110,16 @@ class QuantizedResNeXt101_32x8dWeights(Weights):
100110
"acc@5": 94.480,
101111
},
102112
)
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+
)
103123

104124

105125
def resnet18(

0 commit comments

Comments
 (0)