Skip to content

Commit 69b687d

Browse files
committed
Add zepto weights
1 parent c6e5557 commit 69b687d

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

timm/models/convnext.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -596,10 +596,9 @@ def _cfgv2(url='', **kwargs):
596596
hf_hub_id='timm/',
597597
crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),
598598

599-
'convnext_zepto_rms.untrained': _cfg(
600-
#hf_hub_id='timm/',
601-
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),
602-
test_input_size=(3, 256, 256), test_crop_pct=0.95),
599+
'convnext_zepto_rms.ra4_e3600_r224_in1k': _cfg(
600+
hf_hub_id='timm/',
601+
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)),
603602
'convnext_zepto_rms_ols.untrained': _cfg(
604603
# hf_hub_id='timm/',
605604
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),

0 commit comments

Comments
 (0)