Skip to content

Commit 4a10302

Browse files
committed
Add mobilenet_edgetpu_v2_m weights
1 parent 70182ea commit 4a10302

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

timm/models/efficientnet.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -1762,9 +1762,11 @@ def _cfg(url='', **kwargs):
17621762
'mobilenet_edgetpu_v2_s.untrained': _cfg(
17631763
#hf_hub_id='timm/',
17641764
input_size=(3, 224, 224), crop_pct=0.9),
1765-
'mobilenet_edgetpu_v2_m.untrained': _cfg(
1766-
#hf_hub_id='timm/',
1767-
input_size=(3, 224, 224), crop_pct=0.9),
1765+
'mobilenet_edgetpu_v2_m.ra4_e3600_r224_in1k': _cfg(
1766+
hf_hub_id='timm/',
1767+
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
1768+
crop_pct=0.9, test_input_size=(3, 256, 256), test_crop_pct=0.95,
1769+
),
17681770
'mobilenet_edgetpu_v2_l.untrained': _cfg(
17691771
#hf_hub_id='timm/',
17701772
input_size=(3, 224, 224), crop_pct=0.9),

0 commit comments

Comments
 (0)