yolov5训练最常见错误解决办法

我主要用过两种代码

第一种为哔哩哔哩上的up主Bubbliiiing上传的代码———出现以下错误：

1、标注好自己的数据集之后，进行数据集的划分，通常比例为9：1；

2、接下来进行train.py，一般在改完cls_classes.txt之后基本就可以跑起来了，但是在遇到以下问题时“”一般就是因为你的数据集太少，而且还没有改train.py中的冻结阶段训练参数以及解冻阶段两部分的参数设置；比如我的是121张图片，原始代码的参数为：

但是由于我的数据集太少，所以会出现数据集太少，无法训练；因此我又重新设置了训练次数以及batch_size，主要是为了考虑电脑显存，我改了如下：

此时代码可以进行训练，Epoch为50；

3、运行时会出现运行一代或者几代时会运行中断，如以下的错误：

主要时因为我解决之后才想起来发一下，帮助其他有同样错误的伙伴，所以这个没有截当时pycharm中的图，大家可以对比是不是一样的错；