Skip to content

Commit 237e0b4

Browse files
authored
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
1 parent 743fe1a commit 237e0b4

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

torchvision/prototype/models/efficientnet.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,19 @@ class EfficientNet_B1_Weights(WeightsEnum):
9494
"acc@5": 94.186,
9595
},
9696
)
97-
default = ImageNet1K_V1
97+
ImageNet1K_V2 = Weights(
98+
url="https://download.pytorch.org/models/efficientnet_b1-c27df63c.pth",
99+
transforms=partial(ImageNetEval, crop_size=240, resize_size=255, interpolation=InterpolationMode.BILINEAR),
100+
meta={
101+
**_COMMON_META,
102+
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-lr-wd-crop-tuning",
103+
"interpolation": InterpolationMode.BILINEAR,
104+
"size": (240, 240),
105+
"acc@1": 79.838,
106+
"acc@5": 94.934,
107+
},
108+
)
109+
default = ImageNet1K_V2
98110

99111

100112
class EfficientNet_B2_Weights(WeightsEnum):

0 commit comments

Comments
 (0)