使用traun_aux训练yolov7-e6
报错:indices should be either on cpu or on the same device as the indexed tensor (cpu)
将loss.py第1408,1561
from_which_layer = from_which_layer[fg_mask_inboxes]换为
from_which_layer = from_which_layer.to(fg_mask_inboxes.device)[fg_mask_inboxes]