From 69b687d4cc13e8dcec0f20cdd75bf25d9bda76ab Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 30 Sep 2024 11:38:13 -0700 Subject: [PATCH] Add zepto weights --- timm/models/convnext.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index 9d8b9b94..8c3f31be 100644 --- a/timm/models/convnext.py +++ b/timm/models/convnext.py @@ -596,10 +596,9 @@ default_cfgs = generate_default_cfgs({ hf_hub_id='timm/', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0), - 'convnext_zepto_rms.untrained': _cfg( - #hf_hub_id='timm/', - mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), - test_input_size=(3, 256, 256), test_crop_pct=0.95), + 'convnext_zepto_rms.ra4_e3600_r224_in1k': _cfg( + hf_hub_id='timm/', + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)), 'convnext_zepto_rms_ols.untrained': _cfg( # hf_hub_id='timm/', mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),