Skip to content

Commit f251042

Browse files
committed
mci3/mci4 backbones are stride 64, so 4x4 pooler output
1 parent f242d23 commit f251042

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
@@ -1446,13 +1446,15 @@ def _cfg(url="", **kwargs):
14461446
crop_pct=0.95,
14471447
num_classes=768, # CLIP proj dim
14481448
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
1449+
pool_size=(4, 4),
14491450
license='apple-amlr'
14501451
),
14511452
"fastvit_mci4.apple_mclip2_dfndr2b": _cfg(
14521453
hf_hub_id='timm/',
14531454
crop_pct=0.95,
14541455
num_classes=768, # CLIP proj dim
14551456
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
1457+
pool_size=(4, 4),
14561458
license='apple-amlr'
14571459
),
14581460
})

0 commit comments

Comments
 (0)