dandelin / ViLT

Code for the ICML 2021 (long talk) paper: "ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision"
Apache License 2.0
1.41k stars 208 forks source link

self.mask_token at 553 line in vision_transformer.py #35 I find the self.mask_token has not been defined. #47

Open yr666666 opened 2 years ago

yr666666 commented 2 years ago

Hi, I find that the 'self.mask_token' has not been defined. So I can't run the pre-train code. Can you tell me how to solve it?

Thanks!