playmaker

Hi, still having the problem. I changed that line to equal a torch device of two gpus (passed in set_device). It still runs on 1 gpu.

GPU 0 with about 30000 MiB

I was using a batch size of 6. Should I lower it? Because it says it even on a batch size as low as 2.