大漠yolo5参数求解

遗失的眼泪 · 发表于 2024-4-28 14:51:14

最近刚接触yolo，我的3070显卡头几次训练的时候没问题，多训练几次后就崩溃了，请问训练轮数，批处理，工作线程，在保证识别率准确的情况下，我该降低哪个参数呢？

Bong · 发表于 2024-4-28 15:02:35

不存在崩溃的问题才对

遗失的眼泪 · 发表于 2024-4-28 15:09:50

Bong 发表于 2024-4-28 15:02
不存在崩溃的问题才对

我训练的图片宽度达到1080,300轮数 16批次 4线程，显卡直接崩了。

Bong · 发表于 2024-4-28 15:12:33

遗失的眼泪发表于 2024-4-28 15:09
我训练的图片宽度达到1080,300轮数 16批次 4线程，显卡直接崩了。

不能崩你用的是不是官方的训练不是第三方修改过的吧？第三方修改过的保不准代码有问题

遗失的眼泪 · 发表于 2024-4-28 15:39:27

Bong 发表于 2024-4-28 15:12
不能崩你用的是不是官方的训练不是第三方修改过的吧？第三方修改过的保不准代码有问题 ...

用的是大漠提供的YOLO训练工具，没用原版的

风度猫 · 发表于 2024-4-28 19:04:25

遗失的眼泪发表于 2024-4-28 15:39
用的是大漠提供的YOLO训练工具，没用原版的

直接YY 联系达摩不就好了给你回复的明明白白

遗失的眼泪 · 发表于 2024-4-28 19:55:29

风度猫发表于 2024-4-28 19:04
直接YY 联系达摩不就好了给你回复的明明白白

大漠有时候回复比较慢，所以就想着来论坛先问问，现在崩溃的问题解决了
1.批处理设置14
2.增加虚拟内存
3.关闭百du网盘的自动备份yolo的文件夹，这个会导致训练的时候某个文件没有读写权限导致一直出错

风度猫 · 发表于 2024-4-28 19:56:45

遗失的眼泪发表于 2024-4-28 19:55
大漠有时候回复比较慢，所以就想着来论坛先问问，现在崩溃的问题解决了
1.批处理设置14
2.增加虚拟内存

...估计关键问题是3

		自动登录	找回密码
密码			注册

[求助] 大漠yolo5参数求解