search

roboflow / roboflow-100-benchmark

Code for replicating Roboflow 100 benchmark results and programmatically downloading benchmark datasets
https://www.rf100.org
MIT License
244 stars 23 forks source link

Colab notebook error #51

Closed joy0010 closed 1 year ago

joy0010 commented 1 year ago

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.

截屏2023-07-18 18 10 37
Jacobsolawetz commented 1 year ago

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

https://github.com/WongKinYiu/yolov7/issues/1101#issuecomment-1314659118