Skip to content

GitLab

Unverified Commit 34c8b54f authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Adding new Quantized models. (#4969)

parent b5670284
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 0 deletions
torchvision/prototype/models/quantization/resnet.py
View file @ 34c8b54f
...@@ -87,6 +87,16 @@ class QuantizedResNet50Weights(Weights): ...@@ -87,6 +87,16 @@ class QuantizedResNet50Weights(Weights):
"acc@5": 92.814, "acc@5": 92.814,
}, },
) )
ImageNet1K_FBGEMM_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/quantized/resnet50_fbgemm-23753f79.pth",
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={
**_COMMON_META,
"unquantized": ResNet50Weights.ImageNet1K_RefV2,
"acc@1": 80.282,
"acc@5": 94.976,
},
)
class QuantizedResNeXt101_32x8dWeights(Weights): class QuantizedResNeXt101_32x8dWeights(Weights):
...@@ -100,6 +110,16 @@ class QuantizedResNeXt101_32x8dWeights(Weights): ...@@ -100,6 +110,16 @@ class QuantizedResNeXt101_32x8dWeights(Weights):
"acc@5": 94.480, "acc@5": 94.480,
}, },
) )
ImageNet1K_FBGEMM_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/quantized/resnext101_32x8_fbgemm-ee16d00c.pth",
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={
**_COMMON_META,
"unquantized": ResNeXt101_32x8dWeights.ImageNet1K_RefV2,
"acc@1": 82.574,
"acc@5": 96.132,
},
)
def resnet18( def resnet18(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment