yitu-opensource / T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

'bool' object has no attribute 'pos_embed

helloworld-wu opened this issue · comments

image
When I want to use the pretrained model, I got a problem likes this. Could you tell me how to slove it? Thanks!

把这行改成https://github.com/yitu-opensource/T2T-ViT/blob/main/utils.py#L83

state_dict = load_state_dict(checkpoint_path, model, use_ema, num_classes)

就行了
@helloworld-wu

把这行改成https://github.com/yitu-opensource/T2T-ViT/blob/main/utils.py#L83

state_dict = load_state_dict(checkpoint_path, model, use_ema, num_classes)

就行了 @helloworld-wu

谢谢!