RuntimeError: all tensors must be on devices[0] #118
Description
When I run the code
CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py -c=configs/m2det512_vgg.py --ngpu 4 -t True
There is a error:
RuntimeError: all tensors must be on devices[0]
Does anyone know how to solve it? I tried a lot of methods but they couldn't do it.eg: link Thank you very much!