KeyError: 'Transformer/encoderblock_0\\MultiHeadDotProductAttention_1/query\\kernel is not a file in the archive

[mnauf]

I am trying to train the model using this command "python train_vivit.py --name ucf101 --pretrained_dir /ViT-B_16.npz --dataset custom". It says "KeyError: 'Transformer/encoderblock_0\MultiHeadDotProductAttention_1/query\kernel is not a file in the archive".

Why is that? I have already downloaded the Vit weights., but the path error specified doesn't exist in my project.

[Yuli-yx]

I don't if it's too late but I think the problem is caused by different behaviors of 'os.path.join' under different operating systems. You may simply replace the "\" by "/" manually. Specically, try this:
query_weight = np2th(weights[pjoin(ROOT, ATTENTION_Q, "kernel").replace("\\", "/")]).view(self.hidden_size, self.hidden_size).t(). Do this for every place using 'os.path.join', whch is renamed to ‘pjoin’ in this code.