Usage of GPU

[Einspanner8888]

Hello. I'm jamie from Yonsei Grad school.

I've been researching for Deep survival models using SEER data and i found out yours!
I've tried to make the use of DSM(DeepSruvivalMachines) but, it takes so long to have results due to my heavy data. So, i tried gpu_support branch but it didn't work at all (It was working in only CPU) :(
As far as i'm concerned, DSM is an installed package based on pytorch, which GPU can be supported. I'm wondering whether all branches including 'GPU_support' work on only in CPU?
I'm looking forward to hearing you. Thanks for your sincere efforts.

[Jeanselme]

Hi,

Sorry for the late answer, I have developed a version for GPU, available at Jeanselme/DeepSurvivalMachines - I have also made a push request but feel free to use that version in the meantime and let me know if you encounter any problem
There are two modes: if you indicate cuda = 1, all data will be pushed on GPU, if cuda == 2, only data for the current batch will be pushed (prefer the mode 1 if your data can fit on GPU)