Error with train.py on custom dataset

[wiky231]

Hi, I tried using train.py on a custom Roboflow dataset using the code: %%time %cd /content/ScaledYOLOv4/ !python train.py --img 416 --batch 16 --epochs 100 --data {dataset.location}/data.yaml --cfg ./models/yolov4-csp.yaml --weights '' --name yolov4-csp-results --cache

I got the following error: Optimizer groups: 111 .bias, 115 conv.weight, 108 other Traceback (most recent call last): File "train.py", line 443, in <module> train(hyp, opt, device, tb_writer) File "train.py", line 151, in train world_size=opt.world_size) File "/content/ScaledYOLOv4/utils/datasets.py", line 60, in create_dataloader pad=pad) File "/content/ScaledYOLOv4/utils/datasets.py", line 344, in __init__ labels, shapes = zip(*[cache[x] for x in self.img_files]) TypeError: zip argument #25 must support iteration CPU times: user 89.5 ms, sys: 19 ms, total: 109 ms Wall time: 9.97 s