yitu-opensource/T2T-ViT

'bool' object has no attribute 'pos_embed

helloworld-wu opened this issue · 2 comments

helloworld-wu commented

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

xmy0916 commented

把这行改成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

helloworld-wu commented

把这行改成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

谢谢！