多GPU训练报错

[starsky68]

File "yolov5_all-main/models/yolox.py", line 246, in get_output_and_grid
reg_box[..., :2] = (reg_box[..., :2] +grid)*stride # （第k层）预测物体的中心坐标
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!