search

luyug / GC-DPR

Train Dense Passage Retriever (DPR) with a single GPU
Other
128 stars 20 forks source link

TPU support? #5

Open OhadRubin opened 2 years ago

OhadRubin commented 2 years ago

Will this work on a TPU? If so, can I train it on a colab notebook?

luyug commented 2 years ago

Yes, via Tevatron toolkit