nhadziosma1 issues

Repositories
Issues
Comments

Results 1 issues of


                                            nhadziosma1

Make it possible to choose which GPU to use for training of DPR model, until multi-GPU training gets implemented

Can you please make it possible in the method `train(...)` of the `DensePassageRetriever` class to specify at least the which CUDA-device we want to use. Currently it is hard-coded to...

topic:DPR