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

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.36k stars 209 forks source link

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!