about cuda

[wangxianghan]

Thank you very much for your work, but I have encountered some questions about cuda

[sound]

Because your machine only have two gpu, but I set CUDA_VISIBLE_DEVICES = 3,2,1,0. So you need to change it to "1,0" in train.py (second line). @wangxianghan