Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "712d63c3f06f297e22b1ae32678349187dccd2e4"
Unverified Commit 61399976 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Add new weights for ResNet101 and ResNet151. (#4811)

parent 5b611444
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 0 deletions
torchvision/prototype/models/resnet.py
View file @ 61399976
...@@ -114,6 +114,16 @@ class ResNet101Weights(Weights): ...@@ -114,6 +114,16 @@ class ResNet101Weights(Weights):
"acc@5": 93.546, "acc@5": 93.546,
}, },
) )
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/resnet101-b641f3a9.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,
},
)
class ResNet152Weights(Weights): class ResNet152Weights(Weights):
...@@ -127,6 +137,16 @@ class ResNet152Weights(Weights): ...@@ -127,6 +137,16 @@ class ResNet152Weights(Weights):
"acc@5": 94.046, "acc@5": 94.046,
}, },
) )
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/resnet152-089c0848.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,
},
)
class ResNeXt50_32x4dWeights(Weights): class ResNeXt50_32x4dWeights(Weights):
......
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