Is it running only on CPU?

Open bharatsau opened this issue 5 years ago • 0 comments

Hi, Thanks for the repo. I was trying to run it on DGX server with gpu. I see it is only using CPU. Can you please confirm if I am making any mistake here?

Regards, Bharat Bhusan Sau

Aug 24 '20 12:08 bharatsau