模型训练时报错IndexError: index 10 is out of bounds for dimension 1 with size 10

次次皮

于 2024-02-21 17:44:29 发布

阅读量1.2k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：人工智能学习笔记文章标签：深度学习人工智能机器学习 python

本文链接：https://blog.csdn.net/yszbrzdd/article/details/136217114

报错日志：

      Epoch    GPU_mem   box_loss   cls_loss   dfl_loss  Instances       Size
       1/10         0G      2.561      4.873      1.683          7        640:  98%|█████████▊| 789/808 [11:19<00:16,  1.16it/s]
Traceback (most recent call last):
  File "D:\python\project\illegal-building\train.py", line 17, in <module>
    results = model.train(data="class.yaml", epochs=10, batch=2,device='cpu')  # 训练模型
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Programs\miniconda3\envs\python311\Lib\site-packages\ultralytics\engine\model.py", line 601, in train
    self.trainer.train()
  File "D:\Programs\miniconda3\envs\python311\Lib\site-packages\ultralytics\engine\trainer.py", line 208, in train
    self._do_train(world_size)
  File "D:\Programs\miniconda3\envs\python311\Lib\site-packages\ultralytics\engine\trainer.py", line 376, in _do_train
    self.loss, self.loss_items = self.model(batch)