ezGeno
ezGeno copied to clipboard
Fix cuda arguments for cpu-only device
I got cuda error when running ezGeno in non-gpu server,
even when I set --use_cuda=False
or removing --cuda 0
Some parameters is hard coded or added .cuda()
when cpu-only mode is on,
and here are my minor patch to run it successfully.