Skip to content

Commit d17cd25

Browse files
fmassafacebook-github-bot
authored andcommitted
[fbsync] Add new weights for ResNet101 and ResNet151. (#4811)
Reviewed By: datumbox Differential Revision: D32064684 fbshipit-source-id: f42aec92bb919a7264023406b85df48656c673d5
1 parent 2bbdca8 commit d17cd25

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

torchvision/prototype/models/resnet.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,16 @@ class ResNet101Weights(Weights):
114114
"acc@5": 93.546,
115115
},
116116
)
117+
ImageNet1K_RefV2 = WeightEntry(
118+
url="https://download.pytorch.org/models/resnet101-b641f3a9.pth",
119+
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
120+
meta={
121+
**_common_meta,
122+
"recipe": "https://github.com/pytorch/vision/issues/3995",
123+
"acc@1": 81.728,
124+
"acc@5": 95.670,
125+
},
126+
)
117127

118128

119129
class ResNet152Weights(Weights):
@@ -127,6 +137,16 @@ class ResNet152Weights(Weights):
127137
"acc@5": 94.046,
128138
},
129139
)
140+
ImageNet1K_RefV2 = WeightEntry(
141+
url="https://download.pytorch.org/models/resnet152-089c0848.pth",
142+
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
143+
meta={
144+
**_common_meta,
145+
"recipe": "https://github.com/pytorch/vision/issues/3995",
146+
"acc@1": 82.042,
147+
"acc@5": 95.926,
148+
},
149+
)
130150

131151

132152
class ResNeXt50_32x4dWeights(Weights):

0 commit comments

Comments
 (0)