Amina Hamdi

indices should be either on cpu or on the same device as the indexed tensor (cpu)

11

I'm training yolov7 on a custom dataset using colab: !python train.py --batch 12 --cfg cfg/training/yolov7_custom.yaml --epochs 50 --data data/custom_data.yaml --weights 'yolov7.pt' --device 0 and i face this issue below: Traceback...