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

Add new weights for ResNet101 and ResNet151. (#4811)

parent 5b611444
......@@ -114,6 +114,16 @@ class ResNet101Weights(Weights):
"acc@5": 93.546,
},
)
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/resnet101-b641f3a9.pth",
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={
**_common_meta,
"recipe": "https://github.com/pytorch/vision/issues/3995",
"acc@1": 81.728,
"acc@5": 95.670,
},
)
class ResNet152Weights(Weights):
......@@ -127,6 +137,16 @@ class ResNet152Weights(Weights):
"acc@5": 94.046,
},
)
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/resnet152-089c0848.pth",
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={
**_common_meta,
"recipe": "https://github.com/pytorch/vision/issues/3995",
"acc@1": 82.042,
"acc@5": 95.926,
},
)
class ResNeXt50_32x4dWeights(Weights):
......
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