T2T-ViT icon indicating copy to clipboard operation
T2T-ViT copied to clipboard

'bool' object has no attribute 'pos_embed

Open helloworld-wu opened this issue 4 years ago • 2 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!

helloworld-wu avatar Dec 18 '21 12:12 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

xmy0916 avatar Jan 17 '22 14:01 xmy0916

把这行改成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 avatar Jan 17 '22 14:01 helloworld-wu