mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Take no_emb_class into account when calling resize_pos_embed

Browse Source
This commit is contained in:
Ceshine Lee 2022-07-24 19:11:45 +08:00
parent d7b55a9429
commit 0b64117592
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/vision_transformer.py
View File

@ -644,7 +644,7 @@ def checkpoint_filter_fn(state_dict, model, adapt_layer_scale=False):
v = resize_pos_embed(
v,
model.pos_embed,
getattr(model, 'num_prefix_tokens', 1),
0 if getattr(model, 'no_embed_class') else getattr(model, 'num_prefix_tokens', 1),
model.patch_embed.grid_size
)
elif adapt_layer_scale and 'gamma_' in k: