Colab notebook error

[joy0010]

I followed every step in this video in https://www.youtube.com/watch?v=jIgZMr-PBMo to run the colab code in https://github.com/roboflow/notebooks/blob/main/notebooks/how-to-use-rf100.ipynb.

When I got to this line !python train.py --img 640 --batch 8 --epochs 100 --name $dataset/results --data $dataset/data.yaml --weights 'yolov7_training.pt' |& tee $dataset/logs.txt , it shows RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu).

Please help me to solve this.

[Jacobsolawetz]

Hello @joy0010! This is a bug that has been introduced by the YOLOv7 repository.

Please see this fix that the YOLOv7 community has discovered or consider using a different YOLO version

WongKinYiu/yolov7#1101 (comment)