mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Take `no_emb_class` into account when calling `resize_pos_embed`

pull/1365/head
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 = resize_pos_embed(
v, v,
model.pos_embed, 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 model.patch_embed.grid_size
) )
elif adapt_layer_scale and 'gamma_' in k: elif adapt_layer_scale and 'gamma_' in k: