"...source/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "f663cb2874f918e7663c0daa1d3a3ae8636349ad"
Unverified Commit 237e0b4b authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Add new weights for efficientnet_b1 model (#5033)

* Add new weights for efficientnet_b1 model

* Update accuracies

* Updated the model url and metadata

* Updated default
parent 743fe1ab
......@@ -94,7 +94,19 @@ class EfficientNet_B1_Weights(WeightsEnum):
"acc@5": 94.186,
},
)
default = ImageNet1K_V1
ImageNet1K_V2 = Weights(
url="https://download.pytorch.org/models/efficientnet_b1-c27df63c.pth",
transforms=partial(ImageNetEval, crop_size=240, resize_size=255, interpolation=InterpolationMode.BILINEAR),
meta={
**_COMMON_META,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-lr-wd-crop-tuning",
"interpolation": InterpolationMode.BILINEAR,
"size": (240, 240),
"acc@1": 79.838,
"acc@5": 94.934,
},
)
default = ImageNet1K_V2
class EfficientNet_B2_Weights(WeightsEnum):
......
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