insilicomedicine/GENTRL

How long will it run?

Opened this issue · 1 comments

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??

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.