Skip to content

Commit 97f49d1

Browse files
author
Mehrdad
committed
adding efficientnet_el, efficientnet_es_prunedefficientnet_es_pruned and efficientnet_el_pruned pretrained models to timm/models
1 parent e5ba5dc commit 97f49d1

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

timm/models/efficientnet.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,14 @@ def _cfg(url='', **kwargs):
117117
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_em_ra2-66250f76.pth',
118118
input_size=(3, 240, 240), pool_size=(8, 8), crop_pct=0.882),
119119
'efficientnet_el': _cfg(
120-
url='', input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904),
120+
url='https://github.com/DeGirum/pruned-models/releases/download/efficientnet_v1.0/efficientnet_el.pth',
121+
input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904),
122+
123+
'efficientnet_es_pruned': _cfg(
124+
url='https://github.com/DeGirum/pruned-models/releases/download/efficientnet_v1.0/efficientnet_es_pruned75.pth'),
125+
'efficientnet_el_pruned': _cfg(
126+
url='https://github.com/DeGirum/pruned-models/releases/download/efficientnet_v1.0/efficientnet_el_pruned70.pth',
127+
input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904),
121128

122129
'efficientnet_cc_b0_4e': _cfg(url=''),
123130
'efficientnet_cc_b0_8e': _cfg(url=''),
@@ -1113,6 +1120,12 @@ def efficientnet_es(pretrained=False, **kwargs):
11131120
'efficientnet_es', channel_multiplier=1.0, depth_multiplier=1.0, pretrained=pretrained, **kwargs)
11141121
return model
11151122

1123+
@register_model
1124+
def efficientnet_es_pruned(pretrained=False, **kwargs):
1125+
""" EfficientNet-Edge Small Pruned. For more info: https://github.com/DeGirum/pruned-models/releases/tag/efficientnet_v1.0"""
1126+
model = _gen_efficientnet_edge(
1127+
'efficientnet_es_pruned', channel_multiplier=1.0, depth_multiplier=1.0, pretrained=pretrained, **kwargs)
1128+
return model
11161129

11171130
@register_model
11181131
def efficientnet_em(pretrained=False, **kwargs):
@@ -1129,6 +1142,12 @@ def efficientnet_el(pretrained=False, **kwargs):
11291142
'efficientnet_el', channel_multiplier=1.2, depth_multiplier=1.4, pretrained=pretrained, **kwargs)
11301143
return model
11311144

1145+
@register_model
1146+
def efficientnet_el_pruned(pretrained=False, **kwargs):
1147+
""" EfficientNet-Edge-Large pruned. For more info: https://github.com/DeGirum/pruned-models/releases/tag/efficientnet_v1.0"""
1148+
model = _gen_efficientnet_edge(
1149+
'efficientnet_el_pruned', channel_multiplier=1.2, depth_multiplier=1.4, pretrained=pretrained, **kwargs)
1150+
return model
11321151

11331152
@register_model
11341153
def efficientnet_cc_b0_4e(pretrained=False, **kwargs):

0 commit comments

Comments
 (0)