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
321 stars 37 forks source link

Allow setting device instead of using default device. #34

Open kbraun-axio opened 1 year ago

kbraun-axio commented 1 year ago

On servers with multiple GPUs it is sometimes required to set a specific GPU, like CUDA:1. This is now possible.