Add explicit disable of fc_norm to pe giant spatial variant

This commit is contained in:
Ross Wightman 2025-05-21 07:31:36 -07:00
parent edc37be1c3
commit e7925ea6d6

View File

@ -1678,6 +1678,7 @@ def vit_pe_spatial_gigantic_patch14_448(pretrained=False, **kwargs):
rope_grid_indexing='xy', rope_grid_indexing='xy',
use_pre_transformer_norm=True, use_pre_transformer_norm=True,
use_post_transformer_norm=False, use_post_transformer_norm=False,
use_fc_norm=False, # explicitly disable
init_values=0.1, init_values=0.1,
norm_layer=partial(LayerNorm, eps=1e-5), norm_layer=partial(LayerNorm, eps=1e-5),
#dynamic_img_size=True, #dynamic_img_size=True,