ViT-pytorch
ViT-pytorch copied to clipboard
Key error when loading pre-trained weights
Hi, Thank you for your nice implementation. I get the following error when loading the pre-trained weights:
KeyError: 'Transformer/encoderblock_0\MultiHeadDotProductAttention_1/query\kernel is not a file in the archive'
Would you please help me with this?
Parnian