Traceback (most recent call last):
File "/data/zhongzhijia/code/mindspore/mindyolo/train.py", line 320, in <module>
train(args)
File "/data/zhongzhijia/code/mindspore/mindyolo/train.py", line 275, in train
trainer.train(
File "/data/zhongzhijia/code/mindspore/mindyolo/mindyolo/utils/trainer_factory.py", line 153, in train
for i, data in enumerate(loader):
File "/root/miniconda3/envs/mindspore/lib/python3.9/site-packages/mindspore/dataset/engine/iterators.py", line 152, in __next__
data = self._get_next()
File "/root/miniconda3/envs/mindspore/lib/python3.9/site-packages/mindspore/dataset/engine/iterators.py", line 277, in _get_next
raise err
File "/root/miniconda3/envs/mindspore/lib/python3.9/site-packages/mindspore/dataset/engine/iterators.py", line 260, in _get_next
return {k: self._transform_md_to_output(t) for k, t in self._iterator.GetNextAsMap().items()}
RuntimeError: Exception thrown from user defined Python function in dataset.

------------------------------------------------------------------
- Python Call Stack:
------------------------------------------------------------------
Traceback (most recent call last):
File "/root/miniconda3/envs/mindspore/lib/python3.9/site-packages/mindspore/dataset/engine/datasets_user_defined.py", line 312, in process
result.reraise()
File "/root/miniconda3/envs/mindspore/lib/python3.9/site-packages/mindspore/dataset/core/py_util_helpers.py", line 65, in reraise
raise self.except_type(err_msg)
ValueError: Traceback (most recent call last):
File "/root/miniconda3/envs/mindspore/lib/python3.9/site-packages/mindspore/dataset/engine/datasets_user_defined.py", line 539, in _generator_worker_loop
result = dataset[idx]
File "/data/zhongzhijia/code/mindspore/mindyolo/mindyolo/data/dataset.py", line 313, in __getitem__
sample = getattr(self, func_name)(sample, **_trans)
File "/data/zhongzhijia/code/mindspore/mindyolo/mindyolo/data/dataset.py", line 1076, in label_pad
cls_pad[:min(nL, padding_size)] = cls[:min(nL, padding_size)]
ValueError: could not broadcast input array from shape (15,) into shape (15,1)

------------------------------------------------------------------
- Dataset Pipeline Error Message:
------------------------------------------------------------------
[ERROR] Execute user Python code failed, check 'Python Call Stack' above.

------------------------------------------------------------------
- C++ Call Stack: (For framework developers)
------------------------------------------------------------------
mindspore/ccsrc/minddata/dataset/engine/datasetops/source/generator_op.cc(261).

python train.py --config /data/zhongzhijia/code/mindspore/mindyolo/configs/shwd_yolov8n.yaml --device_target=GPU