Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to use ViLT model for Spanish Text ? #62

Open
karndeepsingh opened this issue Jun 23, 2022 · 1 comment
Open

How to use ViLT model for Spanish Text ? #62

karndeepsingh opened this issue Jun 23, 2022 · 1 comment

Comments

@karndeepsingh
Copy link

Hi,
I have Image and Description of Products which is in Spanish language and want to train a classifier model using ViLT. What kind of pretrained model shall I use to train it on my Spanish Text and Image? I assume the model shared was trained in the English language it won't help me to train on my Spanish text. Correct me If I am wrong!

Please also suggest me the best way to train the model. Shall I train the model from scratch if not then which model weight shall I use to train on my dataset.

Thanks

@altaykacan
Copy link

Hi,

I'm no expert on this and was just browsing the code while studying the paper but I think to be able to use the model for a different language you need to just change the word embedding. Also, the authors mention that in the paper they initialize their transformer with the weights of a pretrained ViT (no language is involved, it's just for visual features).

So just training the model from scratch with the same initial weights as ViT and changing the word embedding should work.

I know it has been some time since you asked your question but I thought it wouldn't hurt to try and answer. Hopefully it was helpful!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants