search

yitu-opensource / T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet
Other
1.15k stars 176 forks source link

Bug in utils.py ->load_for_transfer_learning #63

Open AmnonGeifman opened 3 years ago

AmnonGeifman commented 3 years ago

When using load_state_dict utile function the model needs to be given as input to the function. The first example on the readme doesn't work otherwise.