Take `no_emb_class` into account when calling `resize_pos_embed`

2 years ago · 0b64117592
parent d7b55a9429
commit 0b64117592
1 changed files with 1 additions and 1 deletions
--- a/timm/models/vision_transformer.py
+++ b/timm/models/vision_transformer.py
@ -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: