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

Published 20 hours ago • verified publisher icon jeonsworld

Key error when loading pre-trained weights

Open ParnianA opened this issue 3 years ago • 4 comments

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

ParnianA avatar Jun 10 '21 21:06 ParnianA