Skip to content
This repository has been archived by the owner on Mar 1, 2024. It is now read-only.

Where can I find BLINK's training script? #76

Open
vinhkhuc opened this issue Feb 23, 2021 · 1 comment
Open

Where can I find BLINK's training script? #76

vinhkhuc opened this issue Feb 23, 2021 · 1 comment

Comments

@vinhkhuc
Copy link

Hi BLINK's authors,
Thanks for releasing the pretrained model and the code! They are very useful.
I just wonder where I can find the training script to train the bi-encoder, cross-encoder models on my own data.

ELQ has the training script available at: https://github.com/facebookresearch/BLINK/tree/master/elq_slurm_scripts
Could anyone point me to the corresponding training script for BLINK? I'm looking at the entity linking task for long texts.

Thanks in advance!

@abhinavkulkarni
Copy link

@vinkhuc: It is indeed possible to train a model on your own custom dataset. I would recommend first training a zeshel (Zero-shot learning) model, then bring your data in the same format as zeshel and train your model.

Scripts to download and pre-process zeshel data set are in the repository.

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

No branches or pull requests

2 participants