Skip to content

Commit 913be93

Browse files
committed
Add the experiment accuracy
1 parent da7a18f commit 913be93

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

torchvision/models/regnet.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -583,9 +583,8 @@ class RegNet_Y_16GF_Weights(WeightsEnum):
583583
meta={
584584
**_COMMON_SWAG_META,
585585
"num_params": 83590140,
586-
# still mock
587-
"acc@1": 86.02,
588-
"acc@5": 98.05,
586+
"acc@1": 86.012,
587+
"acc@5": 98.054,
589588
},
590589
)
591590
DEFAULT = IMAGENET1K_V2
@@ -622,9 +621,8 @@ class RegNet_Y_32GF_Weights(WeightsEnum):
622621
meta={
623622
**_COMMON_SWAG_META,
624623
"num_params": 145046770,
625-
# still mock
626-
"acc@1": 86.83,
627-
"acc@5": 98.36,
624+
"acc@1": 86.838,
625+
"acc@5": 98.362,
628626
},
629627
)
630628
DEFAULT = IMAGENET1K_V2

0 commit comments

Comments
 (0)