yoloV5报错，看一下广播问题

Question

Traceback (most recent call last):
  File "D:/xiangmu/yolov5.0/train.py", line 545, in <module>
    train(hyp, opt, device, tb_writer)
  File "D:/xiangmu/yolov5.0/train.py", line 280, in train
    for i, (imgs, targets, paths, _) in pbar:  # batch -------------------------------------------------------------
  File "D:\aaa\envs\pao\lib\site-packages\tqdm\std.py", line 1195, in __iter__
    for obj in iterable:
  File "D:\xiangmu\yolov5.0\utils\datasets.py", line 104, in __iter__
    yield next(self.iterator)
  File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\dataloader.py", line 530, in __next__
    data = self._next_data()
  File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\dataloader.py", line 570, in _next_data
    data = self._dataset_fetcher.fetch(index)  # may raise StopIteration
  File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "D:\xiangmu\yolov5.0\utils\datasets.py", line 525, in __getitem__
    img, labels = load_mosaic(self, index)
  File "D:\xiangmu\yolov5.0\utils\datasets.py", line 698, in load_mosaic
    img4[y1a:y2a, x1a:x2a] = img[y1b:y2b, x1b:x2b]  # img4[ymin:ymax, xmin:xmax]
ValueError: could not broadcast input array from shape (404,640,4) into shape (404,640,3)