Skip to content

Commit 20b98d6

Browse files
kazhangdatumbox
authored andcommitted
[fbsync] Add new weights for wide_resnet101_2 model (#4906)
Summary: * Add new weights for wide_resnet101_2 model * Updated accuracies Reviewed By: datumbox Differential Revision: D32470481 fbshipit-source-id: 145391d20ca8c6da3cee44239a1df93eb350d733 Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent fe00a8d commit 20b98d6

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)