Vision-Transformer

Vision-Transformer copied to clipboard

Inconsistent name 'ViT' and Missing file 'transformers'

2

Hi there! Thanks for this repository The code doesn't run though... Two issues that I found: 1. The model name in the file is ViT, while in the training file...

sptom

I sincerely hope that the author can complete the code

kongweiming

After running line 134 in the ViT forward: patches = images.unfold(2, self.patch_height, self.patch_width).unfold(3, self.patch_height, self.patch_width) I get a tensor with sizes  The next line is: patches = patches.permute(0, 2,...

sptom