Skip to content

GitLab

"src/diffusers/commands/diffusers_cli.py" did not exist on "bf13b76aa31845ac3f55ec7e8a445a6c74a86af9"
Unverified Commit c5fb79f8 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Add new weights for ResNeXt50_32x4d. (#4836)

parent 9b7d5d52
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 0 deletions
torchvision/prototype/models/resnet.py
View file @ c5fb79f8
...@@ -160,6 +160,16 @@ class ResNeXt50_32x4dWeights(Weights): ...@@ -160,6 +160,16 @@ class ResNeXt50_32x4dWeights(Weights):
"acc@5": 93.698, "acc@5": 93.698,
}, },
) )
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/resnext50_32x4d-b260af35.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,
},
)
class ResNeXt101_32x8dWeights(Weights): class ResNeXt101_32x8dWeights(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