Skip to content

Commit 2d1bc8a

Browse files
committed
First conv is different for fastvit mci3/mci4 models
1 parent 639d6ab commit 2d1bc8a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

timm/models/fastvit.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1447,6 +1447,7 @@ def _cfg(url="", **kwargs):
14471447
num_classes=768, # CLIP proj dim
14481448
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
14491449
pool_size=(4, 4),
1450+
first_conv='stem.0.conv_kxk.0.conv',
14501451
license='apple-amlr'
14511452
),
14521453
"fastvit_mci4.apple_mclip2_dfndr2b": _cfg(
@@ -1455,6 +1456,7 @@ def _cfg(url="", **kwargs):
14551456
num_classes=768, # CLIP proj dim
14561457
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
14571458
pool_size=(4, 4),
1459+
first_conv='stem.0.conv_kxk.0.conv',
14581460
license='apple-amlr'
14591461
),
14601462
})

0 commit comments

Comments
 (0)