Skip to content

Commit bb1ab47

Browse files
authored
Fix Accuracy meta-data on shufflenetv2 (#5896)
1 parent d07da41 commit bb1ab47

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torchvision/models/shufflenetv2.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -198,8 +198,8 @@ class ShuffleNet_V2_X0_5_Weights(WeightsEnum):
198198
**_COMMON_META,
199199
"num_params": 1366792,
200200
"metrics": {
201-
"acc@1": 69.362,
202-
"acc@5": 88.316,
201+
"acc@1": 60.552,
202+
"acc@5": 81.746,
203203
},
204204
},
205205
)
@@ -214,8 +214,8 @@ class ShuffleNet_V2_X1_0_Weights(WeightsEnum):
214214
**_COMMON_META,
215215
"num_params": 2278604,
216216
"metrics": {
217-
"acc@1": 60.552,
218-
"acc@5": 81.746,
217+
"acc@1": 69.362,
218+
"acc@5": 88.316,
219219
},
220220
},
221221
)

0 commit comments

Comments
 (0)