search

UKPLab / gpl

Powerful unsupervised domain adaptation method for dense retrieval. Requires only unlabeled corpus and yields massive improvement: "GPL: Generative Pseudo Labeling for Unsupervised Domain Adaptation of Dense Retrieval" https://arxiv.org/abs/2112.07577
Apache License 2.0
315 stars 39 forks source link

GPU speedup #8

Closed ahadda5 closed 2 years ago

ahadda5 commented 2 years ago

I recon this is more of a generic question for TSADE + GPL (or any transformer used) , but can you use GPU by simply doing something like gpl.to(device)?

ahadda5 commented 2 years ago

already supported [2022-06-06 10:29:03] INFO [beir.generation.models.auto_model.init:16] Use pytorch device: cuda