mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Add explicit disable of fc_norm to pe giant spatial variant
This commit is contained in:
parent
edc37be1c3
commit
e7925ea6d6
@ -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,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user