From e0c17d7716cdda7b53a21dee4a1e99b54592ac5c Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 10 Nov 2021 12:31:46 +0000 Subject: [PATCH 01/10] Add new weights for MobileNetV3 Large. --- torchvision/prototype/models/mobilenetv3.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/torchvision/prototype/models/mobilenetv3.py b/torchvision/prototype/models/mobilenetv3.py index 8e1fad903a0..d89c9e36d5c 100644 --- a/torchvision/prototype/models/mobilenetv3.py +++ b/torchvision/prototype/models/mobilenetv3.py @@ -41,7 +41,6 @@ def _mobilenet_v3( "size": (224, 224), "categories": _IMAGENET_CATEGORIES, "interpolation": InterpolationMode.BILINEAR, - "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#mobilenetv3-large--small", } @@ -51,10 +50,21 @@ class MobileNetV3LargeWeights(Weights): transforms=partial(ImageNetEval, crop_size=224), meta={ **_common_meta, + "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#mobilenetv3-large--small", "acc@1": 74.042, "acc@5": 91.340, }, ) + ImageNet1K_RefV2 = WeightEntry( + url="https://download.pytorch.org/models/mobilenet_v3_large-d7307c3f.pth", + transforms=partial(ImageNetEval, crop_size=224, resize_size=232), + meta={ + **_common_meta, + "recipe": "https://github.com/pytorch/vision/issues/3995", + "acc@1": 75.152, + "acc@5": 92.634, + }, + ) class MobileNetV3SmallWeights(Weights): @@ -63,6 +73,7 @@ class MobileNetV3SmallWeights(Weights): transforms=partial(ImageNetEval, crop_size=224), meta={ **_common_meta, + "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#mobilenetv3-large--small", "acc@1": 67.668, "acc@5": 87.402, }, From dae33e7529f49c55473574240545f47f7086ee2f Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 10 Nov 2021 13:17:13 +0000 Subject: [PATCH 02/10] Update weights for ResNet101. --- torchvision/prototype/models/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index ff82d5e3172..4edfa238571 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -115,13 +115,13 @@ class ResNet101Weights(Weights): }, ) ImageNet1K_RefV2 = WeightEntry( - url="https://download.pytorch.org/models/resnet101-b641f3a9.pth", + url="https://download.pytorch.org/models/resnet101-cd907fc2.pth", transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 81.728, - "acc@5": 95.670, + "acc@1": 81.890, + "acc@5": 95.776, }, ) From 112f89e14bb415860afa509257ebf763e373f0d8 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 10 Nov 2021 13:28:49 +0000 Subject: [PATCH 03/10] Update weights for ResNet152. --- torchvision/prototype/models/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index 4edfa238571..edd58146d3f 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -138,13 +138,13 @@ class ResNet152Weights(Weights): }, ) ImageNet1K_RefV2 = WeightEntry( - url="https://download.pytorch.org/models/resnet152-089c0848.pth", + url="https://download.pytorch.org/models/resnet152-f82ba261.pth", transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 82.042, - "acc@5": 95.926, + "acc@1": 82.286, + "acc@5": 96.008, }, ) From a406d176e9868b4562f8f9cd5ee1b4d6ffcf2325 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 10 Nov 2021 15:24:57 +0000 Subject: [PATCH 04/10] Update numbers of ResNet101 and ResNet152 with batch size 1. --- torchvision/prototype/models/resnet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index edd58146d3f..c5962b6ee77 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -120,8 +120,8 @@ class ResNet101Weights(Weights): meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 81.890, - "acc@5": 95.776, + "acc@1": 81.886, + "acc@5": 95.780, }, ) @@ -143,8 +143,8 @@ class ResNet152Weights(Weights): meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 82.286, - "acc@5": 96.008, + "acc@1": 82.284, + "acc@5": 96.002, }, ) From 2bbbbbb93b2030ddbd05bfd91deb4e33c314ee07 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 10 Nov 2021 16:00:20 +0000 Subject: [PATCH 05/10] Adding new weights for ResNeXt101_32x8d --- torchvision/prototype/models/resnet.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index c5962b6ee77..c114918ac74 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -183,6 +183,16 @@ class ResNeXt101_32x8dWeights(Weights): "acc@5": 94.526, }, ) + ImageNet1K_RefV2 = WeightEntry( + url="https://download.pytorch.org/models/resnext101_32x8d-110c445d.pth", + transforms=partial(ImageNetEval, crop_size=224, resize_size=232), + meta={ + **_common_meta, + "recipe": "https://github.com/pytorch/vision/issues/3995", + "acc@1": 82.284, + "acc@5": 96.002, + }, + ) class WideResNet50_2Weights(Weights): From e844cc7dbca9ae03b71d609174daf4b464de2a58 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 10 Nov 2021 17:22:26 +0000 Subject: [PATCH 06/10] Batch size 1 stats for ResNeXt101_32x8d --- torchvision/prototype/models/resnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index c114918ac74..7d4eeb1b493 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -189,8 +189,8 @@ class ResNeXt101_32x8dWeights(Weights): meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 82.284, - "acc@5": 96.002, + "acc@1": 82.834, + "acc@5": 96.228, }, ) From a0a0e82242090825cdaa1174fccd764ee2b1560f Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Mon, 15 Nov 2021 13:08:27 +0000 Subject: [PATCH 07/10] Update weights for MobileNetV3 Large. --- torchvision/prototype/models/mobilenetv3.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/prototype/models/mobilenetv3.py b/torchvision/prototype/models/mobilenetv3.py index c953a8a7b0e..c73b255259c 100644 --- a/torchvision/prototype/models/mobilenetv3.py +++ b/torchvision/prototype/models/mobilenetv3.py @@ -56,13 +56,13 @@ class MobileNetV3LargeWeights(Weights): }, ) ImageNet1K_RefV2 = WeightEntry( - url="https://download.pytorch.org/models/mobilenet_v3_large-d7307c3f.pth", + url="https://download.pytorch.org/models/mobilenet_v3_large-5c1a4163.pth", transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 75.152, - "acc@5": 92.634, + "acc@1": 75.274, + "acc@5": 92.566, }, ) From 1cf517cfec5d8be1909fd42ebabd1630f6326e18 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Mon, 15 Nov 2021 14:30:10 +0000 Subject: [PATCH 08/10] Update weights for ResNeXt50. --- torchvision/prototype/models/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index e7b3e95d104..6a3fe26856f 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -161,13 +161,13 @@ class ResNeXt50_32x4dWeights(Weights): }, ) ImageNet1K_RefV2 = WeightEntry( - url="https://download.pytorch.org/models/resnext50_32x4d-b260af35.pth", + url="https://download.pytorch.org/models/resnext50_32x4d-1a0047aa.pth", transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 81.116, - "acc@5": 95.478, + "acc@1": 81.198, + "acc@5": 95.340, }, ) From 9b767773f14cb67a15da2c88cb2e4b4a2dacc39d Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 16 Nov 2021 15:57:11 +0000 Subject: [PATCH 09/10] Fix merge issues. --- torchvision/prototype/models/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index 53fda40bc34..9dc6a061cc4 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -187,7 +187,7 @@ class ResNeXt101_32x8dWeights(Weights): url="https://download.pytorch.org/models/resnext101_32x8d-110c445d.pth", transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ - **_common_meta, + **_COMMON_META, "recipe": "https://github.com/pytorch/vision/issues/3995", "acc@1": 82.834, "acc@5": 96.228, From 30ec7ec848c0a3d46ff13f3560a4af10ce0ea41f Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 16 Nov 2021 16:15:57 +0000 Subject: [PATCH 10/10] Updating the URLs to indicate the recipe config. --- torchvision/prototype/models/mobilenetv3.py | 2 +- torchvision/prototype/models/resnet.py | 14 +++++++------- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/torchvision/prototype/models/mobilenetv3.py b/torchvision/prototype/models/mobilenetv3.py index 4c3b6eba664..ff05ce7848a 100644 --- a/torchvision/prototype/models/mobilenetv3.py +++ b/torchvision/prototype/models/mobilenetv3.py @@ -60,7 +60,7 @@ class MobileNetV3LargeWeights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-reg-tuning", "acc@1": 75.274, "acc@5": 92.566, }, diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index 9dc6a061cc4..f3003cb99b9 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -96,7 +96,7 @@ class ResNet50Weights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "acc@1": 80.674, "acc@5": 95.166, }, @@ -119,7 +119,7 @@ class ResNet101Weights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "acc@1": 81.886, "acc@5": 95.780, }, @@ -142,7 +142,7 @@ class ResNet152Weights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "acc@1": 82.284, "acc@5": 96.002, }, @@ -165,7 +165,7 @@ class ResNeXt50_32x4dWeights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "acc@1": 81.198, "acc@5": 95.340, }, @@ -188,7 +188,7 @@ class ResNeXt101_32x8dWeights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "acc@1": 82.834, "acc@5": 96.228, }, @@ -211,7 +211,7 @@ class WideResNet50_2Weights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "acc@1": 81.602, "acc@5": 95.758, }, @@ -234,7 +234,7 @@ class WideResNet101_2Weights(Weights): transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META, - "recipe": "https://github.com/pytorch/vision/issues/3995", + "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "acc@1": 82.492, "acc@5": 96.110, },