search

sebastian-hofstaetter / matchmaker

Training & evaluation library for text-based neural re-ranking and dense retrieval models built with PyTorch
https://neural-ir-explorer.ec.tuwien.ac.at/
Apache License 2.0
261 stars 30 forks source link

Pre-trained model #4

Open subhashiniy opened 4 years ago

subhashiniy commented 4 years ago

Hi, Are there pre-trained models available for download?

Thanks in advance.

sebastian-hofstaetter commented 4 years ago

Hi, I am sorry at the moment we don't have pre-trained models to download. But TK is quite efficient to train from scratch - if you have a single newer GPU it will only take a couple of hours to get very good results. Best, Sebastian

seanmacavaney commented 4 years ago

Any chance you could reconsider and release the pre-trained models?

sebastian-hofstaetter commented 4 years ago

Yeah I guess so :) which models would you be interested in? The original TK/TKL or the knowledge distilled ones?

seanmacavaney commented 4 years ago

Awesome- thanks! The original TK and TKL would be great.

abheesht17 commented 3 years ago

Hello! Have the pretrained models for TK been released? If not, could you please share them? Would be of great help! :)