Skip to content

GitLab

Unverified Commit 2ec0e847 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

New schema for metrics in weights meta-data (#6047) 

* Classif models

* Detection

* Segmentation

* quantization

* Video

* optical flow

* tests

* Fix docs

* Fix Video dataset

* Consistency for RAFT dataset names

* use ImageNet-1K

* Use COCO-val2017-VOC-labels for segmentation

* formatting
parent 2a35dde3
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 495 additions and 297 deletions
torchvision/models/quantization/inception.py
View file @ 2ec0e847
...@@ -183,9 +183,11 @@ class Inception_V3_QuantizedWeights(WeightsEnum): ...@@ -183,9 +183,11 @@ class Inception_V3_QuantizedWeights(WeightsEnum):
"backend": "fbgemm", "backend": "fbgemm",
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-models", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-models",
"unquantized": Inception_V3_Weights.IMAGENET1K_V1, "unquantized": Inception_V3_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 77.176, "ImageNet-1K": {
"acc@5": 93.354, "acc@1": 77.176,
"acc@5": 93.354,
}
}, },
"_docs": """ "_docs": """
These weights were produced by doing Post Training Quantization (eager mode) on top of the unquantized These weights were produced by doing Post Training Quantization (eager mode) on top of the unquantized
......
torchvision/models/quantization/mobilenetv2.py
View file @ 2ec0e847
...@@ -75,9 +75,11 @@ class MobileNet_V2_QuantizedWeights(WeightsEnum): ...@@ -75,9 +75,11 @@ class MobileNet_V2_QuantizedWeights(WeightsEnum):
"backend": "qnnpack", "backend": "qnnpack",
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#qat-mobilenetv2", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#qat-mobilenetv2",
"unquantized": MobileNet_V2_Weights.IMAGENET1K_V1, "unquantized": MobileNet_V2_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 71.658, "ImageNet-1K": {
"acc@5": 90.150, "acc@1": 71.658,
"acc@5": 90.150,
}
}, },
"_docs": """ "_docs": """
These weights were produced by doing Quantization Aware Training (eager mode) on top of the unquantized These weights were produced by doing Quantization Aware Training (eager mode) on top of the unquantized
......
torchvision/models/quantization/mobilenetv3.py
View file @ 2ec0e847
...@@ -169,9 +169,11 @@ class MobileNet_V3_Large_QuantizedWeights(WeightsEnum): ...@@ -169,9 +169,11 @@ class MobileNet_V3_Large_QuantizedWeights(WeightsEnum):
"backend": "qnnpack", "backend": "qnnpack",
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#qat-mobilenetv3", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#qat-mobilenetv3",
"unquantized": MobileNet_V3_Large_Weights.IMAGENET1K_V1, "unquantized": MobileNet_V3_Large_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 73.004, "ImageNet-1K": {
"acc@5": 90.858, "acc@1": 73.004,
"acc@5": 90.858,
}
}, },
"_docs": """ "_docs": """
These weights were produced by doing Quantization Aware Training (eager mode) on top of the unquantized These weights were produced by doing Quantization Aware Training (eager mode) on top of the unquantized
......
torchvision/models/quantization/resnet.py
View file @ 2ec0e847
...@@ -169,9 +169,11 @@ class ResNet18_QuantizedWeights(WeightsEnum): ...@@ -169,9 +169,11 @@ class ResNet18_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 11689512, "num_params": 11689512,
"unquantized": ResNet18_Weights.IMAGENET1K_V1, "unquantized": ResNet18_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 69.494, "ImageNet-1K": {
"acc@5": 88.882, "acc@1": 69.494,
"acc@5": 88.882,
}
}, },
}, },
) )
...@@ -186,9 +188,11 @@ class ResNet50_QuantizedWeights(WeightsEnum): ...@@ -186,9 +188,11 @@ class ResNet50_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 25557032, "num_params": 25557032,
"unquantized": ResNet50_Weights.IMAGENET1K_V1, "unquantized": ResNet50_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 75.920, "ImageNet-1K": {
"acc@5": 92.814, "acc@1": 75.920,
"acc@5": 92.814,
}
}, },
}, },
) )
...@@ -199,9 +203,11 @@ class ResNet50_QuantizedWeights(WeightsEnum): ...@@ -199,9 +203,11 @@ class ResNet50_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 25557032, "num_params": 25557032,
"unquantized": ResNet50_Weights.IMAGENET1K_V2, "unquantized": ResNet50_Weights.IMAGENET1K_V2,
"metrics": { "_metrics": {
"acc@1": 80.282, "ImageNet-1K": {
"acc@5": 94.976, "acc@1": 80.282,
"acc@5": 94.976,
}
}, },
}, },
) )
...@@ -216,9 +222,11 @@ class ResNeXt101_32X8D_QuantizedWeights(WeightsEnum): ...@@ -216,9 +222,11 @@ class ResNeXt101_32X8D_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 88791336, "num_params": 88791336,
"unquantized": ResNeXt101_32X8D_Weights.IMAGENET1K_V1, "unquantized": ResNeXt101_32X8D_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 78.986, "ImageNet-1K": {
"acc@5": 94.480, "acc@1": 78.986,
"acc@5": 94.480,
}
}, },
}, },
) )
...@@ -229,9 +237,11 @@ class ResNeXt101_32X8D_QuantizedWeights(WeightsEnum): ...@@ -229,9 +237,11 @@ class ResNeXt101_32X8D_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 88791336, "num_params": 88791336,
"unquantized": ResNeXt101_32X8D_Weights.IMAGENET1K_V2, "unquantized": ResNeXt101_32X8D_Weights.IMAGENET1K_V2,
"metrics": { "_metrics": {
"acc@1": 82.574, "ImageNet-1K": {
"acc@5": 96.132, "acc@1": 82.574,
"acc@5": 96.132,
}
}, },
}, },
) )
...@@ -247,9 +257,11 @@ class ResNeXt101_64X4D_QuantizedWeights(WeightsEnum): ...@@ -247,9 +257,11 @@ class ResNeXt101_64X4D_QuantizedWeights(WeightsEnum):
"num_params": 83455272, "num_params": 83455272,
"recipe": "https://github.com/pytorch/vision/pull/5935", "recipe": "https://github.com/pytorch/vision/pull/5935",
"unquantized": ResNeXt101_64X4D_Weights.IMAGENET1K_V1, "unquantized": ResNeXt101_64X4D_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 82.898, "ImageNet-1K": {
"acc@5": 96.326, "acc@1": 82.898,
"acc@5": 96.326,
}
}, },
}, },
) )
......
torchvision/models/quantization/shufflenetv2.py
View file @ 2ec0e847
...@@ -133,9 +133,11 @@ class ShuffleNet_V2_X0_5_QuantizedWeights(WeightsEnum): ...@@ -133,9 +133,11 @@ class ShuffleNet_V2_X0_5_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 1366792, "num_params": 1366792,
"unquantized": ShuffleNet_V2_X0_5_Weights.IMAGENET1K_V1, "unquantized": ShuffleNet_V2_X0_5_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 57.972, "ImageNet-1K": {
"acc@5": 79.780, "acc@1": 57.972,
"acc@5": 79.780,
}
}, },
}, },
) )
...@@ -150,9 +152,11 @@ class ShuffleNet_V2_X1_0_QuantizedWeights(WeightsEnum): ...@@ -150,9 +152,11 @@ class ShuffleNet_V2_X1_0_QuantizedWeights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 2278604, "num_params": 2278604,
"unquantized": ShuffleNet_V2_X1_0_Weights.IMAGENET1K_V1, "unquantized": ShuffleNet_V2_X1_0_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 68.360, "ImageNet-1K": {
"acc@5": 87.582, "acc@1": 68.360,
"acc@5": 87.582,
}
}, },
}, },
) )
...@@ -168,9 +172,11 @@ class ShuffleNet_V2_X1_5_QuantizedWeights(WeightsEnum): ...@@ -168,9 +172,11 @@ class ShuffleNet_V2_X1_5_QuantizedWeights(WeightsEnum):
"recipe": "https://github.com/pytorch/vision/pull/5906", "recipe": "https://github.com/pytorch/vision/pull/5906",
"num_params": 3503624, "num_params": 3503624,
"unquantized": ShuffleNet_V2_X1_5_Weights.IMAGENET1K_V1, "unquantized": ShuffleNet_V2_X1_5_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 72.052, "ImageNet-1K": {
"acc@5": 90.700, "acc@1": 72.052,
"acc@5": 90.700,
}
}, },
}, },
) )
...@@ -186,9 +192,11 @@ class ShuffleNet_V2_X2_0_QuantizedWeights(WeightsEnum): ...@@ -186,9 +192,11 @@ class ShuffleNet_V2_X2_0_QuantizedWeights(WeightsEnum):
"recipe": "https://github.com/pytorch/vision/pull/5906", "recipe": "https://github.com/pytorch/vision/pull/5906",
"num_params": 7393996, "num_params": 7393996,
"unquantized": ShuffleNet_V2_X2_0_Weights.IMAGENET1K_V1, "unquantized": ShuffleNet_V2_X2_0_Weights.IMAGENET1K_V1,
"metrics": { "_metrics": {
"acc@1": 75.354, "ImageNet-1K": {
"acc@5": 92.488, "acc@1": 75.354,
"acc@5": 92.488,
}
}, },
}, },
) )
......
torchvision/models/regnet.py
View file @ 2ec0e847
This diff is collapsed.
torchvision/models/resnet.py
View file @ 2ec0e847
...@@ -317,9 +317,11 @@ class ResNet18_Weights(WeightsEnum): ...@@ -317,9 +317,11 @@ class ResNet18_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 11689512, "num_params": 11689512,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet",
"metrics": { "_metrics": {
"acc@1": 69.758, "ImageNet-1K": {
"acc@5": 89.078, "acc@1": 69.758,
"acc@5": 89.078,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -335,9 +337,11 @@ class ResNet34_Weights(WeightsEnum): ...@@ -335,9 +337,11 @@ class ResNet34_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 21797672, "num_params": 21797672,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet",
"metrics": { "_metrics": {
"acc@1": 73.314, "ImageNet-1K": {
"acc@5": 91.420, "acc@1": 73.314,
"acc@5": 91.420,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -353,9 +357,11 @@ class ResNet50_Weights(WeightsEnum): ...@@ -353,9 +357,11 @@ class ResNet50_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 25557032, "num_params": 25557032,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet",
"metrics": { "_metrics": {
"acc@1": 76.130, "ImageNet-1K": {
"acc@5": 92.862, "acc@1": 76.130,
"acc@5": 92.862,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -367,9 +373,11 @@ class ResNet50_Weights(WeightsEnum): ...@@ -367,9 +373,11 @@ class ResNet50_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 25557032, "num_params": 25557032,
"recipe": "https://github.com/pytorch/vision/issues/3995#issuecomment-1013906621", "recipe": "https://github.com/pytorch/vision/issues/3995#issuecomment-1013906621",
"metrics": { "_metrics": {
"acc@1": 80.858, "ImageNet-1K": {
"acc@5": 95.434, "acc@1": 80.858,
"acc@5": 95.434,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
...@@ -388,9 +396,11 @@ class ResNet101_Weights(WeightsEnum): ...@@ -388,9 +396,11 @@ class ResNet101_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 44549160, "num_params": 44549160,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet",
"metrics": { "_metrics": {
"acc@1": 77.374, "ImageNet-1K": {
"acc@5": 93.546, "acc@1": 77.374,
"acc@5": 93.546,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -402,9 +412,11 @@ class ResNet101_Weights(WeightsEnum): ...@@ -402,9 +412,11 @@ class ResNet101_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 44549160, "num_params": 44549160,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe",
"metrics": { "_metrics": {
"acc@1": 81.886, "ImageNet-1K": {
"acc@5": 95.780, "acc@1": 81.886,
"acc@5": 95.780,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
...@@ -423,9 +435,11 @@ class ResNet152_Weights(WeightsEnum): ...@@ -423,9 +435,11 @@ class ResNet152_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 60192808, "num_params": 60192808,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnet",
"metrics": { "_metrics": {
"acc@1": 78.312, "ImageNet-1K": {
"acc@5": 94.046, "acc@1": 78.312,
"acc@5": 94.046,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -437,9 +451,11 @@ class ResNet152_Weights(WeightsEnum): ...@@ -437,9 +451,11 @@ class ResNet152_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 60192808, "num_params": 60192808,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe",
"metrics": { "_metrics": {
"acc@1": 82.284, "ImageNet-1K": {
"acc@5": 96.002, "acc@1": 82.284,
"acc@5": 96.002,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
...@@ -458,9 +474,11 @@ class ResNeXt50_32X4D_Weights(WeightsEnum): ...@@ -458,9 +474,11 @@ class ResNeXt50_32X4D_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 25028904, "num_params": 25028904,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnext", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnext",
"metrics": { "_metrics": {
"acc@1": 77.618, "ImageNet-1K": {
"acc@5": 93.698, "acc@1": 77.618,
"acc@5": 93.698,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -472,9 +490,11 @@ class ResNeXt50_32X4D_Weights(WeightsEnum): ...@@ -472,9 +490,11 @@ class ResNeXt50_32X4D_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 25028904, "num_params": 25028904,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe",
"metrics": { "_metrics": {
"acc@1": 81.198, "ImageNet-1K": {
"acc@5": 95.340, "acc@1": 81.198,
"acc@5": 95.340,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
...@@ -493,9 +513,11 @@ class ResNeXt101_32X8D_Weights(WeightsEnum): ...@@ -493,9 +513,11 @@ class ResNeXt101_32X8D_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 88791336, "num_params": 88791336,
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnext", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#resnext",
"metrics": { "_metrics": {
"acc@1": 79.312, "ImageNet-1K": {
"acc@5": 94.526, "acc@1": 79.312,
"acc@5": 94.526,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -507,9 +529,11 @@ class ResNeXt101_32X8D_Weights(WeightsEnum): ...@@ -507,9 +529,11 @@ class ResNeXt101_32X8D_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 88791336, "num_params": 88791336,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres",
"metrics": { "_metrics": {
"acc@1": 82.834, "ImageNet-1K": {
"acc@5": 96.228, "acc@1": 82.834,
"acc@5": 96.228,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
...@@ -528,9 +552,11 @@ class ResNeXt101_64X4D_Weights(WeightsEnum): ...@@ -528,9 +552,11 @@ class ResNeXt101_64X4D_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 83455272, "num_params": 83455272,
"recipe": "https://github.com/pytorch/vision/pull/5935", "recipe": "https://github.com/pytorch/vision/pull/5935",
"metrics": { "_metrics": {
"acc@1": 83.246, "ImageNet-1K": {
"acc@5": 96.454, "acc@1": 83.246,
"acc@5": 96.454,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using TorchVision's `new training recipe These weights were trained from scratch by using TorchVision's `new training recipe
...@@ -549,9 +575,11 @@ class Wide_ResNet50_2_Weights(WeightsEnum): ...@@ -549,9 +575,11 @@ class Wide_ResNet50_2_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 68883240, "num_params": 68883240,
"recipe": "https://github.com/pytorch/vision/pull/912#issue-445437439", "recipe": "https://github.com/pytorch/vision/pull/912#issue-445437439",
"metrics": { "_metrics": {
"acc@1": 78.468, "ImageNet-1K": {
"acc@5": 94.086, "acc@1": 78.468,
"acc@5": 94.086,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -563,9 +591,11 @@ class Wide_ResNet50_2_Weights(WeightsEnum): ...@@ -563,9 +591,11 @@ class Wide_ResNet50_2_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 68883240, "num_params": 68883240,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres",
"metrics": { "_metrics": {
"acc@1": 81.602, "ImageNet-1K": {
"acc@5": 95.758, "acc@1": 81.602,
"acc@5": 95.758,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
...@@ -584,9 +614,11 @@ class Wide_ResNet101_2_Weights(WeightsEnum): ...@@ -584,9 +614,11 @@ class Wide_ResNet101_2_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 126886696, "num_params": 126886696,
"recipe": "https://github.com/pytorch/vision/pull/912#issue-445437439", "recipe": "https://github.com/pytorch/vision/pull/912#issue-445437439",
"metrics": { "_metrics": {
"acc@1": 78.848, "ImageNet-1K": {
"acc@5": 94.284, "acc@1": 78.848,
"acc@5": 94.284,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""", "_docs": """These weights reproduce closely the results of the paper using a simple training recipe.""",
}, },
...@@ -598,9 +630,11 @@ class Wide_ResNet101_2_Weights(WeightsEnum): ...@@ -598,9 +630,11 @@ class Wide_ResNet101_2_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 126886696, "num_params": 126886696,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe", "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe",
"metrics": { "_metrics": {
"acc@1": 82.510, "ImageNet-1K": {
"acc@5": 96.020, "acc@1": 82.510,
"acc@5": 96.020,
}
}, },
"_docs": """ "_docs": """
These weights improve upon the results of the original paper by using TorchVision's `new training recipe These weights improve upon the results of the original paper by using TorchVision's `new training recipe
......
torchvision/models/segmentation/deeplabv3.py
View file @ 2ec0e847
...@@ -146,9 +146,11 @@ class DeepLabV3_ResNet50_Weights(WeightsEnum): ...@@ -146,9 +146,11 @@ class DeepLabV3_ResNet50_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 42004074, "num_params": 42004074,
"recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_resnet50", "recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_resnet50",
"metrics": { "_metrics": {
"miou": 66.4, "COCO-val2017-VOC-labels": {
"pixel_acc": 92.4, "miou": 66.4,
"pixel_acc": 92.4,
}
}, },
}, },
) )
...@@ -163,9 +165,11 @@ class DeepLabV3_ResNet101_Weights(WeightsEnum): ...@@ -163,9 +165,11 @@ class DeepLabV3_ResNet101_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 60996202, "num_params": 60996202,
"recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#fcn_resnet101", "recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#fcn_resnet101",
"metrics": { "_metrics": {
"miou": 67.4, "COCO-val2017-VOC-labels": {
"pixel_acc": 92.4, "miou": 67.4,
"pixel_acc": 92.4,
}
}, },
}, },
) )
...@@ -180,9 +184,11 @@ class DeepLabV3_MobileNet_V3_Large_Weights(WeightsEnum): ...@@ -180,9 +184,11 @@ class DeepLabV3_MobileNet_V3_Large_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 11029328, "num_params": 11029328,
"recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_mobilenet_v3_large", "recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_mobilenet_v3_large",
"metrics": { "_metrics": {
"miou": 60.3, "COCO-val2017-VOC-labels": {
"pixel_acc": 91.2, "miou": 60.3,
"pixel_acc": 91.2,
}
}, },
}, },
) )
......
torchvision/models/segmentation/fcn.py
View file @ 2ec0e847
...@@ -65,9 +65,11 @@ class FCN_ResNet50_Weights(WeightsEnum): ...@@ -65,9 +65,11 @@ class FCN_ResNet50_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 35322218, "num_params": 35322218,
"recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#fcn_resnet50", "recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#fcn_resnet50",
"metrics": { "_metrics": {
"miou": 60.5, "COCO-val2017-VOC-labels": {
"pixel_acc": 91.4, "miou": 60.5,
"pixel_acc": 91.4,
}
}, },
}, },
) )
...@@ -82,9 +84,11 @@ class FCN_ResNet101_Weights(WeightsEnum): ...@@ -82,9 +84,11 @@ class FCN_ResNet101_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"num_params": 54314346, "num_params": 54314346,
"recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_resnet101", "recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_resnet101",
"metrics": { "_metrics": {
"miou": 63.7, "COCO-val2017-VOC-labels": {
"pixel_acc": 91.9, "miou": 63.7,
"pixel_acc": 91.9,
}
}, },
}, },
) )
......
torchvision/models/segmentation/lraspp.py
View file @ 2ec0e847
...@@ -102,9 +102,11 @@ class LRASPP_MobileNet_V3_Large_Weights(WeightsEnum): ...@@ -102,9 +102,11 @@ class LRASPP_MobileNet_V3_Large_Weights(WeightsEnum):
"categories": _VOC_CATEGORIES, "categories": _VOC_CATEGORIES,
"min_size": (1, 1), "min_size": (1, 1),
"recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#lraspp_mobilenet_v3_large", "recipe": "https://github.com/pytorch/vision/tree/main/references/segmentation#lraspp_mobilenet_v3_large",
"metrics": { "_metrics": {
"miou": 57.9, "COCO-val2017-VOC-labels": {
"pixel_acc": 91.2, "miou": 57.9,
"pixel_acc": 91.2,
}
}, },
"_docs": """ "_docs": """
These weights were trained on a subset of COCO, using only the 20 categories that are present in the These weights were trained on a subset of COCO, using only the 20 categories that are present in the
......
torchvision/models/shufflenetv2.py
View file @ 2ec0e847
...@@ -198,9 +198,11 @@ class ShuffleNet_V2_X0_5_Weights(WeightsEnum): ...@@ -198,9 +198,11 @@ class ShuffleNet_V2_X0_5_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 1366792, "num_params": 1366792,
"metrics": { "_metrics": {
"acc@1": 60.552, "ImageNet-1K": {
"acc@5": 81.746, "acc@1": 60.552,
"acc@5": 81.746,
}
}, },
"_docs": """These weights were trained from scratch to reproduce closely the results of the paper.""", "_docs": """These weights were trained from scratch to reproduce closely the results of the paper.""",
}, },
...@@ -216,9 +218,11 @@ class ShuffleNet_V2_X1_0_Weights(WeightsEnum): ...@@ -216,9 +218,11 @@ class ShuffleNet_V2_X1_0_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 2278604, "num_params": 2278604,
"metrics": { "_metrics": {
"acc@1": 69.362, "ImageNet-1K": {
"acc@5": 88.316, "acc@1": 69.362,
"acc@5": 88.316,
}
}, },
"_docs": """These weights were trained from scratch to reproduce closely the results of the paper.""", "_docs": """These weights were trained from scratch to reproduce closely the results of the paper.""",
}, },
...@@ -234,9 +238,11 @@ class ShuffleNet_V2_X1_5_Weights(WeightsEnum): ...@@ -234,9 +238,11 @@ class ShuffleNet_V2_X1_5_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"recipe": "https://github.com/pytorch/vision/pull/5906", "recipe": "https://github.com/pytorch/vision/pull/5906",
"num_params": 3503624, "num_params": 3503624,
"metrics": { "_metrics": {
"acc@1": 72.996, "ImageNet-1K": {
"acc@5": 91.086, "acc@1": 72.996,
"acc@5": 91.086,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using TorchVision's `new training recipe These weights were trained from scratch by using TorchVision's `new training recipe
...@@ -255,9 +261,11 @@ class ShuffleNet_V2_X2_0_Weights(WeightsEnum): ...@@ -255,9 +261,11 @@ class ShuffleNet_V2_X2_0_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"recipe": "https://github.com/pytorch/vision/pull/5906", "recipe": "https://github.com/pytorch/vision/pull/5906",
"num_params": 7393996, "num_params": 7393996,
"metrics": { "_metrics": {
"acc@1": 76.230, "ImageNet-1K": {
"acc@5": 93.006, "acc@1": 76.230,
"acc@5": 93.006,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using TorchVision's `new training recipe These weights were trained from scratch by using TorchVision's `new training recipe
......
torchvision/models/squeezenet.py
View file @ 2ec0e847
...@@ -129,9 +129,11 @@ class SqueezeNet1_0_Weights(WeightsEnum): ...@@ -129,9 +129,11 @@ class SqueezeNet1_0_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"min_size": (21, 21), "min_size": (21, 21),
"num_params": 1248424, "num_params": 1248424,
"metrics": { "_metrics": {
"acc@1": 58.092, "ImageNet-1K": {
"acc@5": 80.420, "acc@1": 58.092,
"acc@5": 80.420,
}
}, },
}, },
) )
...@@ -146,9 +148,11 @@ class SqueezeNet1_1_Weights(WeightsEnum): ...@@ -146,9 +148,11 @@ class SqueezeNet1_1_Weights(WeightsEnum):
**_COMMON_META, **_COMMON_META,
"min_size": (17, 17), "min_size": (17, 17),
"num_params": 1235496, "num_params": 1235496,
"metrics": { "_metrics": {
"acc@1": 58.178, "ImageNet-1K": {
"acc@5": 80.624, "acc@1": 58.178,
"acc@5": 80.624,
}
}, },
}, },
) )
......
torchvision/models/swin_transformer.py
View file @ 2ec0e847
...@@ -417,9 +417,11 @@ class Swin_T_Weights(WeightsEnum): ...@@ -417,9 +417,11 @@ class Swin_T_Weights(WeightsEnum):
"num_params": 28288354, "num_params": 28288354,
"min_size": (224, 224), "min_size": (224, 224),
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#swintransformer", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#swintransformer",
"metrics": { "_metrics": {
"acc@1": 81.358, "ImageNet-1K": {
"acc@5": 95.526, "acc@1": 81.358,
"acc@5": 95.526,
}
}, },
"_docs": """These weights reproduce closely the results of the paper using its training recipe.""", "_docs": """These weights reproduce closely the results of the paper using its training recipe.""",
}, },
......
torchvision/models/vgg.py
View file @ 2ec0e847
...@@ -121,9 +121,11 @@ class VGG11_Weights(WeightsEnum): ...@@ -121,9 +121,11 @@ class VGG11_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 132863336, "num_params": 132863336,
"metrics": { "_metrics": {
"acc@1": 69.020, "ImageNet-1K": {
"acc@5": 88.628, "acc@1": 69.020,
"acc@5": 88.628,
}
}, },
}, },
) )
...@@ -137,9 +139,11 @@ class VGG11_BN_Weights(WeightsEnum): ...@@ -137,9 +139,11 @@ class VGG11_BN_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 132868840, "num_params": 132868840,
"metrics": { "_metrics": {
"acc@1": 70.370, "ImageNet-1K": {
"acc@5": 89.810, "acc@1": 70.370,
"acc@5": 89.810,
}
}, },
}, },
) )
...@@ -153,9 +157,11 @@ class VGG13_Weights(WeightsEnum): ...@@ -153,9 +157,11 @@ class VGG13_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 133047848, "num_params": 133047848,
"metrics": { "_metrics": {
"acc@1": 69.928, "ImageNet-1K": {
"acc@5": 89.246, "acc@1": 69.928,
"acc@5": 89.246,
}
}, },
}, },
) )
...@@ -169,9 +175,11 @@ class VGG13_BN_Weights(WeightsEnum): ...@@ -169,9 +175,11 @@ class VGG13_BN_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 133053736, "num_params": 133053736,
"metrics": { "_metrics": {
"acc@1": 71.586, "ImageNet-1K": {
"acc@5": 90.374, "acc@1": 71.586,
"acc@5": 90.374,
}
}, },
}, },
) )
...@@ -185,9 +193,11 @@ class VGG16_Weights(WeightsEnum): ...@@ -185,9 +193,11 @@ class VGG16_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 138357544, "num_params": 138357544,
"metrics": { "_metrics": {
"acc@1": 71.592, "ImageNet-1K": {
"acc@5": 90.382, "acc@1": 71.592,
"acc@5": 90.382,
}
}, },
}, },
) )
...@@ -205,9 +215,11 @@ class VGG16_Weights(WeightsEnum): ...@@ -205,9 +215,11 @@ class VGG16_Weights(WeightsEnum):
"num_params": 138357544, "num_params": 138357544,
"categories": None, "categories": None,
"recipe": "https://github.com/amdegroot/ssd.pytorch#training-ssd", "recipe": "https://github.com/amdegroot/ssd.pytorch#training-ssd",
"metrics": { "_metrics": {
"acc@1": float("nan"), "ImageNet-1K": {
"acc@5": float("nan"), "acc@1": float("nan"),
"acc@5": float("nan"),
}
}, },
"_docs": """ "_docs": """
These weights can't be used for classification because they are missing values in the `classifier` These weights can't be used for classification because they are missing values in the `classifier`
...@@ -226,9 +238,11 @@ class VGG16_BN_Weights(WeightsEnum): ...@@ -226,9 +238,11 @@ class VGG16_BN_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 138365992, "num_params": 138365992,
"metrics": { "_metrics": {
"acc@1": 73.360, "ImageNet-1K": {
"acc@5": 91.516, "acc@1": 73.360,
"acc@5": 91.516,
}
}, },
}, },
) )
...@@ -242,9 +256,11 @@ class VGG19_Weights(WeightsEnum): ...@@ -242,9 +256,11 @@ class VGG19_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 143667240, "num_params": 143667240,
"metrics": { "_metrics": {
"acc@1": 72.376, "ImageNet-1K": {
"acc@5": 90.876, "acc@1": 72.376,
"acc@5": 90.876,
}
}, },
}, },
) )
...@@ -258,9 +274,11 @@ class VGG19_BN_Weights(WeightsEnum): ...@@ -258,9 +274,11 @@ class VGG19_BN_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 143678248, "num_params": 143678248,
"metrics": { "_metrics": {
"acc@1": 74.218, "ImageNet-1K": {
"acc@5": 91.842, "acc@1": 74.218,
"acc@5": 91.842,
}
}, },
}, },
) )
......
torchvision/models/video/resnet.py
View file @ 2ec0e847
...@@ -323,9 +323,11 @@ class R3D_18_Weights(WeightsEnum): ...@@ -323,9 +323,11 @@ class R3D_18_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 33371472, "num_params": 33371472,
"metrics": { "_metrics": {
"acc@1": 52.75, "Kinetics-400": {
"acc@5": 75.45, "acc@1": 52.75,
"acc@5": 75.45,
}
}, },
}, },
) )
...@@ -339,9 +341,11 @@ class MC3_18_Weights(WeightsEnum): ...@@ -339,9 +341,11 @@ class MC3_18_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 11695440, "num_params": 11695440,
"metrics": { "_metrics": {
"acc@1": 53.90, "Kinetics-400": {
"acc@5": 76.29, "acc@1": 53.90,
"acc@5": 76.29,
}
}, },
}, },
) )
...@@ -355,9 +359,11 @@ class R2Plus1D_18_Weights(WeightsEnum): ...@@ -355,9 +359,11 @@ class R2Plus1D_18_Weights(WeightsEnum):
meta={ meta={
**_COMMON_META, **_COMMON_META,
"num_params": 31505325, "num_params": 31505325,
"metrics": { "_metrics": {
"acc@1": 57.50, "Kinetics-400": {
"acc@5": 78.81, "acc@1": 57.50,
"acc@5": 78.81,
}
}, },
}, },
) )
......
torchvision/models/vision_transformer.py
View file @ 2ec0e847
...@@ -328,9 +328,11 @@ class ViT_B_16_Weights(WeightsEnum): ...@@ -328,9 +328,11 @@ class ViT_B_16_Weights(WeightsEnum):
"num_params": 86567656, "num_params": 86567656,
"min_size": (224, 224), "min_size": (224, 224),
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_b_16", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_b_16",
"metrics": { "_metrics": {
"acc@1": 81.072, "ImageNet-1K": {
"acc@5": 95.318, "acc@1": 81.072,
"acc@5": 95.318,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using a modified version of `DeIT These weights were trained from scratch by using a modified version of `DeIT
...@@ -350,9 +352,11 @@ class ViT_B_16_Weights(WeightsEnum): ...@@ -350,9 +352,11 @@ class ViT_B_16_Weights(WeightsEnum):
**_COMMON_SWAG_META, **_COMMON_SWAG_META,
"num_params": 86859496, "num_params": 86859496,
"min_size": (384, 384), "min_size": (384, 384),
"metrics": { "_metrics": {
"acc@1": 85.304, "ImageNet-1K": {
"acc@5": 97.650, "acc@1": 85.304,
"acc@5": 97.650,
}
}, },
"_docs": """ "_docs": """
These weights are learnt via transfer learning by end-to-end fine-tuning the original These weights are learnt via transfer learning by end-to-end fine-tuning the original
...@@ -373,9 +377,11 @@ class ViT_B_16_Weights(WeightsEnum): ...@@ -373,9 +377,11 @@ class ViT_B_16_Weights(WeightsEnum):
"recipe": "https://github.com/pytorch/vision/pull/5793", "recipe": "https://github.com/pytorch/vision/pull/5793",
"num_params": 86567656, "num_params": 86567656,
"min_size": (224, 224), "min_size": (224, 224),
"metrics": { "_metrics": {
"acc@1": 81.886, "ImageNet-1K": {
"acc@5": 96.180, "acc@1": 81.886,
"acc@5": 96.180,
}
}, },
"_docs": """ "_docs": """
These weights are composed of the original frozen `SWAG <https://arxiv.org/abs/2201.08371>`_ trunk These weights are composed of the original frozen `SWAG <https://arxiv.org/abs/2201.08371>`_ trunk
...@@ -395,9 +401,11 @@ class ViT_B_32_Weights(WeightsEnum): ...@@ -395,9 +401,11 @@ class ViT_B_32_Weights(WeightsEnum):
"num_params": 88224232, "num_params": 88224232,
"min_size": (224, 224), "min_size": (224, 224),
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_b_32", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_b_32",
"metrics": { "_metrics": {
"acc@1": 75.912, "ImageNet-1K": {
"acc@5": 92.466, "acc@1": 75.912,
"acc@5": 92.466,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using a modified version of `DeIT These weights were trained from scratch by using a modified version of `DeIT
...@@ -417,9 +425,11 @@ class ViT_L_16_Weights(WeightsEnum): ...@@ -417,9 +425,11 @@ class ViT_L_16_Weights(WeightsEnum):
"num_params": 304326632, "num_params": 304326632,
"min_size": (224, 224), "min_size": (224, 224),
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_l_16", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_l_16",
"metrics": { "_metrics": {
"acc@1": 79.662, "ImageNet-1K": {
"acc@5": 94.638, "acc@1": 79.662,
"acc@5": 94.638,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using a modified version of TorchVision's These weights were trained from scratch by using a modified version of TorchVision's
...@@ -440,9 +450,11 @@ class ViT_L_16_Weights(WeightsEnum): ...@@ -440,9 +450,11 @@ class ViT_L_16_Weights(WeightsEnum):
**_COMMON_SWAG_META, **_COMMON_SWAG_META,
"num_params": 305174504, "num_params": 305174504,
"min_size": (512, 512), "min_size": (512, 512),
"metrics": { "_metrics": {
"acc@1": 88.064, "ImageNet-1K": {
"acc@5": 98.512, "acc@1": 88.064,
"acc@5": 98.512,
}
}, },
"_docs": """ "_docs": """
These weights are learnt via transfer learning by end-to-end fine-tuning the original These weights are learnt via transfer learning by end-to-end fine-tuning the original
...@@ -463,9 +475,11 @@ class ViT_L_16_Weights(WeightsEnum): ...@@ -463,9 +475,11 @@ class ViT_L_16_Weights(WeightsEnum):
"recipe": "https://github.com/pytorch/vision/pull/5793", "recipe": "https://github.com/pytorch/vision/pull/5793",
"num_params": 304326632, "num_params": 304326632,
"min_size": (224, 224), "min_size": (224, 224),
"metrics": { "_metrics": {
"acc@1": 85.146, "ImageNet-1K": {
"acc@5": 97.422, "acc@1": 85.146,
"acc@5": 97.422,
}
}, },
"_docs": """ "_docs": """
These weights are composed of the original frozen `SWAG <https://arxiv.org/abs/2201.08371>`_ trunk These weights are composed of the original frozen `SWAG <https://arxiv.org/abs/2201.08371>`_ trunk
...@@ -485,9 +499,11 @@ class ViT_L_32_Weights(WeightsEnum): ...@@ -485,9 +499,11 @@ class ViT_L_32_Weights(WeightsEnum):
"num_params": 306535400, "num_params": 306535400,
"min_size": (224, 224), "min_size": (224, 224),
"recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_l_32", "recipe": "https://github.com/pytorch/vision/tree/main/references/classification#vit_l_32",
"metrics": { "_metrics": {
"acc@1": 76.972, "ImageNet-1K": {
"acc@5": 93.07, "acc@1": 76.972,
"acc@5": 93.07,
}
}, },
"_docs": """ "_docs": """
These weights were trained from scratch by using a modified version of `DeIT These weights were trained from scratch by using a modified version of `DeIT
...@@ -511,9 +527,11 @@ class ViT_H_14_Weights(WeightsEnum): ...@@ -511,9 +527,11 @@ class ViT_H_14_Weights(WeightsEnum):
**_COMMON_SWAG_META, **_COMMON_SWAG_META,
"num_params": 633470440, "num_params": 633470440,
"min_size": (518, 518), "min_size": (518, 518),
"metrics": { "_metrics": {
"acc@1": 88.552, "ImageNet-1K": {
"acc@5": 98.694, "acc@1": 88.552,
"acc@5": 98.694,
}
}, },
"_docs": """ "_docs": """
These weights are learnt via transfer learning by end-to-end fine-tuning the original These weights are learnt via transfer learning by end-to-end fine-tuning the original
...@@ -534,9 +552,11 @@ class ViT_H_14_Weights(WeightsEnum): ...@@ -534,9 +552,11 @@ class ViT_H_14_Weights(WeightsEnum):
"recipe": "https://github.com/pytorch/vision/pull/5793", "recipe": "https://github.com/pytorch/vision/pull/5793",
"num_params": 632045800, "num_params": 632045800,
"min_size": (224, 224), "min_size": (224, 224),
"metrics": { "_metrics": {
"acc@1": 85.708, "ImageNet-1K": {
"acc@5": 97.730, "acc@1": 85.708,
"acc@5": 97.730,
}
}, },
"_docs": """ "_docs": """
These weights are composed of the original frozen `SWAG <https://arxiv.org/abs/2201.08371>`_ trunk These weights are composed of the original frozen `SWAG <https://arxiv.org/abs/2201.08371>`_ trunk
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment