Fix Accuracy meta-data on shufflenetv2 (#5896)

bb1ab475 · Vasilis Vryniotis · GitHub · d07da41d · bb1ab475
Unverified Commit bb1ab475 authored Apr 27, 2022 by Vasilis Vryniotis Committed by GitHub Apr 27, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

torchvision/models/shufflenetv2.py torchvision/models/shufflenetv2.py +4 -4

No files found.
--- a/torchvision/models/shufflenetv2.py
+++ b/torchvision/models/shufflenetv2.py
@@ -198,8 +198,8 @@ class ShuffleNet_V2_X0_5_Weights(WeightsEnum):
            **_COMMON_META,
            "num_params": 1366792,
            "metrics": {
-                "acc@1": 69.362,
-                "acc@5": 88.316,
+                "acc@1": 60.552,
+                "acc@5": 81.746,
            },
        },
    )
@@ -214,8 +214,8 @@ class ShuffleNet_V2_X1_0_Weights(WeightsEnum):
            **_COMMON_META,
            "num_params": 2278604,
            "metrics": {
-                "acc@1": 60.552,
-                "acc@5": 81.746,
+                "acc@1": 69.362,
+                "acc@5": 88.316,
            },
        },
    )