TypeError: cannot unpack non-iterable int object

[pantment]

I train the model in custom data and in the last epochs it breaks with this error:

ERROR in training steps. ERROR in training loop or eval/save model. Traceback (most recent call last): File "/media/FastData/pmentesidis/yolov6/YOLOv6/tools/train.py", line 142, in main(args) File "/media/FastData/pmentesidis/yolov6/YOLOv6/tools/train.py", line 132, in main trainer.train() File "/media/FastData/pmentesidis/yolov6/YOLOv6/yolov6/core/engine.py", line 121, in train self.train_one_epoch(self.epoch) File "/media/FastData/pmentesidis/yolov6/YOLOv6/yolov6/core/engine.py", line 134, in train_one_epoch for self.step, self.batch_data in self.pbar: File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/tqdm/std.py", line 1178, in iter for obj in iterable: File "/media/FastData/pmentesidis/yolov6/YOLOv6/yolov6/data/data_load.py", line 111, in iter yield next(self.iterator) File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 633, in next data = self._next_data() File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_data return self._process_data(data) File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_data data.reraise() File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/_utils.py", line 644, in reraise raise exception TypeError: Caught TypeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/media/FastData/pmentesidis/anaconda3/envs/yolov6/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/media/FastData/pmentesidis/yolov6/YOLOv6/yolov6/data/datasets.py", line 164, in getitem img, labels = random_affine( File "/media/FastData/pmentesidis/yolov6/YOLOv6/yolov6/data/data_augment.py", line 82, in random_affine height, width = new_shape TypeError: cannot unpack non-iterable int object

Additional

No response

[mtjhl]

What's your training commands?

[pantment]

python /media/FastDatayolov6/YOLOv6/tools/train.py --batch 20 --conf /media/FastData/yolov6/YOLOv6/configs/yolov6m6_finetune.py --data /media/FastData/yolov6/YOLOv6/20230629-new_dataset/data.yaml --device 0,1 --name 20230629yolo6m6