"!yolo task=detect mode=train model=yolov5l6u.pt data={dataset.location}/data.yaml epochs=100 imgsz=640" not running any epochs

[jeffnguyen27]

The above command stopped right after setup, before running any epoch. I also tried running with yolov5x.pt and yolov516.pt, with no difference. I'm running this on colab. Any idea how to fix this? Thanks.

[Agares9]

use GPU

[Khush1593]

!python train.py --img 640 --batch 10 --epochs 100 --data /content/tennis-ball-detection-6/data.yaml --weights /content/yolov5l6.pt --name yolov5l6_finetune

Also You can use this code for training. This is python code so it works well in CPU as well as in GPU. Here you have to change the value of variables according to your requirements.