How long will it run? - Githubissues

insilicomedicine / GENTRL

Generative Tensorial Reinforcement Learning (GENTRL) model

611 stars 218 forks source link

How long will it run? #25

Open cgh2797 opened 4 years ago

cgh2797 commented 4 years ago

hi now i running pretrain.ipynb with GPU(RTX 2060)

How long will it run?

and i also wondering play time of train_rl.ipynb

may be one month??

Bibyutatsu commented 4 years ago

Hi, It will take around 1-2 day depending on the number of epochs. I want to tell you I have transformed this model into a Pytorch Lightning module for Multi GPU support. Please check it out here

It will increase your efficiency of running it on multi-GPU and even a single GPU.