Skip to content

Commit 348f75c

Browse files
Add new weights for wide_resnet101_2 model (#4906)
* Add new weights for wide_resnet101_2 model * Updated accuracies Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent a2d240e commit 348f75c

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

torchvision/prototype/models/resnet.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,16 @@ class WideResNet101_2Weights(Weights):
219219
"acc@5": 94.284,
220220
},
221221
)
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+
)
222232

223233

224234
def resnet18(weights: Optional[ResNet18Weights] = None, progress: bool = True, **kwargs: Any) -> ResNet:

0 commit comments

Comments
 (0)