multi GPU capabilities - Githubissues

adjidieng / ETM

Topic Modeling in Embedding Spaces

MIT License

540 stars 127 forks source link

multi GPU capabilities #7

Open thousandoaks opened 4 years ago

thousandoaks commented 4 years ago

Hi, really interesting project. Are there plans to make it multi-GPU ?

adjidieng commented 4 years ago

Hi there,

There is currently no plans to add code for multi-GPU. However I believe that is easily doable on your end using PyTorch. Let us know if you have other questions.