Question

我正在google colab中使用yolo darknet训练模型，但是当我开始训练时，页面冻结，并且出现一个弹出窗口，表明网页没有响应我不知道是不是因为该模型需要训练许多类并且页面崩溃了

这是我的代码：

!apt-get update
!unzip "/content/drive/My Drive/custom_dib_model/darknet.zip"
!sudo apt install dos2unix
!find . -type f -print0 | xargs -0 dos2unix
!chmod +x /content/darknet
!make
!./darknet detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights data/person.jpg
!rm /content/darknet/backup -r
!ln -s /content/drive/'My Drive'/dib_weights/backup /content/darknet
!./darknet detector train dibujos_dataset/dib.data dib_yolov4.cfg yolov4.conv.137 -map -dont_show

最后一行是开始训练我的模型的那一行，当页面冻结时大约经过了5分钟，应该注意没有出现错误

我找到了类似的问题，但没有具体答案 possible answer由用户

这是我能提供给您的所有信息，我希望它足够了

Answer 1

我通过更改YOLO配置dib_yolo.cfg文件（将细分从64修改为16）解决了该问题

当我开始训练模型时，YOLO暗网冻结

1 个答案: