File tree 1 file changed +20
-0
lines changed
torchvision/prototype/models 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -114,6 +114,16 @@ class ResNet101Weights(Weights):
114
114
"acc@5" : 93.546 ,
115
115
},
116
116
)
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
+ )
117
127
118
128
119
129
class ResNet152Weights (Weights ):
@@ -127,6 +137,16 @@ class ResNet152Weights(Weights):
127
137
"acc@5" : 94.046 ,
128
138
},
129
139
)
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
+ )
130
150
131
151
132
152
class ResNeXt50_32x4dWeights (Weights ):
You can’t perform that action at this time.
0 commit comments