We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fe00a8d commit 20b98d6Copy full SHA for 20b98d6
torchvision/prototype/models/resnet.py
@@ -219,6 +219,16 @@ class WideResNet101_2Weights(Weights):
219
"acc@5": 94.284,
220
},
221
)
222
+ ImageNet1K_RefV2 = WeightEntry(
223
+ url="https://download.pytorch.org/models/wide_resnet101_2-b8680a8c.pth",
224
+ transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
225
+ meta={
226
+ **_common_meta,
227
+ "recipe": "https://github.com/pytorch/vision/issues/3995",
228
+ "acc@1": 82.492,
229
+ "acc@5": 96.110,
230
+ },
231
+ )
232
233
234
def resnet18(weights: Optional[ResNet18Weights] = None, progress: bool = True, **kwargs: Any) -> ResNet:
0 commit comments