Skip to content

Commit 88b7ef6

Browse files
committed
Disable dynamic_img_size default on PE models for now
1 parent fa7287f commit 88b7ef6

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

timm/models/eva.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1565,7 +1565,7 @@ def vit_pe_core_base_patch16_224(pretrained=False, **kwargs):
15651565
attn_pool_num_heads=8,
15661566
attn_pool_mlp_ratio=4.,
15671567
norm_layer=partial(LayerNorm, eps=1e-5),
1568-
dynamic_img_size=True
1568+
#dynamic_img_size=True
15691569
)
15701570
return _create_eva('vit_pe_core_base_patch16_224', pretrained=pretrained, **dict(model_args, **kwargs))
15711571

@@ -1588,7 +1588,7 @@ def vit_pe_core_large_patch14_336(pretrained=False, **kwargs):
15881588
attn_pool_num_heads=8,
15891589
attn_pool_mlp_ratio=4.,
15901590
norm_layer=partial(LayerNorm, eps=1e-5),
1591-
dynamic_img_size=True,
1591+
#dynamic_img_size=True,
15921592
)
15931593
return _create_eva('vit_pe_core_large_patch14_336', pretrained=pretrained, **dict(model_args, **kwargs))
15941594

@@ -1611,7 +1611,7 @@ def vit_pe_core_gigantic_patch14_448(pretrained=False, **kwargs):
16111611
attn_pool_num_heads=8,
16121612
attn_pool_mlp_ratio=4.,
16131613
norm_layer=partial(LayerNorm, eps=1e-5),
1614-
dynamic_img_size=True,
1614+
#dynamic_img_size=True,
16151615
)
16161616
return _create_eva('vit_pe_core_gigantic_patch14_448', pretrained=pretrained, **dict(model_args, **kwargs))
16171617

@@ -1635,7 +1635,7 @@ def vit_pe_lang_large_patch14_448(pretrained=False, **kwargs):
16351635
use_fc_norm=False, # explicitly disable
16361636
init_values=0.1,
16371637
norm_layer=partial(LayerNorm, eps=1e-5),
1638-
dynamic_img_size=True,
1638+
#dynamic_img_size=True,
16391639
)
16401640
return _create_eva('vit_pe_lang_large_patch14_448', pretrained=pretrained, **dict(model_args, **kwargs))
16411641

@@ -1658,7 +1658,7 @@ def vit_pe_lang_gigantic_patch14_448(pretrained=False, **kwargs):
16581658
use_fc_norm=False, # explicitly disable
16591659
init_values=0.1,
16601660
norm_layer=partial(LayerNorm, eps=1e-5),
1661-
dynamic_img_size=True,
1661+
#dynamic_img_size=True,
16621662
)
16631663
return _create_eva('vit_pe_lang_gigantic_patch14_448', pretrained=pretrained, **dict(model_args, **kwargs))
16641664

@@ -1680,7 +1680,7 @@ def vit_pe_spatial_gigantic_patch14_448(pretrained=False, **kwargs):
16801680
use_post_transformer_norm=False,
16811681
init_values=0.1,
16821682
norm_layer=partial(LayerNorm, eps=1e-5),
1683-
dynamic_img_size=True,
1683+
#dynamic_img_size=True,
16841684
)
16851685
return _create_eva('vit_pe_spatial_gigantic_patch14_448', pretrained=pretrained, **dict(model_args, **kwargs))
16861686

0 commit comments

Comments
 (0)