在使用yolo训练的时候出现以下错误RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

这个错误通常是因为索引张量（indices）和被索引的张量（indexed tensor）不在同一个设备上。要解决这个问题，可以将索引张量（indices）移动到与被索引的张量（indexed tensor）相同的设备上，例如：

# 将索引张量（indices）移动到与被索引的张量（indexed tensor）相同的设备上
indices = indices.to(device)
output = indexed_tensor[indices]

其中 `device` 是被索引的张量（indexed tensor）所在的设备，可以通过 `indexed_tensor.device` 获得。