search

insilicomedicine / GENTRL

Generative Tensorial Reinforcement Learning (GENTRL) model
611 stars 218 forks source link

Multi GPU support #18

Closed Bibyutatsu closed 4 years ago

Bibyutatsu commented 5 years ago

Hi, Can you please help me to understand how can I use multi GPU to train your model? I want to know whether the simple torch.DataParallel(gentrl) will work or there is some additional things I have to do.

Bibyutatsu commented 4 years ago

Hi, All who are following this thread, I want to tell you I have transformed this model into a Pytorch Lightning module for Multi GPU support. Please check it out here

https://github.com/Bibyutatsu/proGENTRL