From 2d054053d4ae038a07bed63c12cd564c0615a172 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 27 Apr 2022 11:45:04 +0100 Subject: [PATCH] Fix Accuracy meta-data on shufflenetv2 --- torchvision/models/shufflenetv2.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/torchvision/models/shufflenetv2.py b/torchvision/models/shufflenetv2.py index 77f07438b20..68243e06e8f 100644 --- 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, }, }, )