热门标签
热门文章
- 1【粉丝福利社】AI时代Python量化交易实战：ChatGPT让量化交易插上翅膀（文末送书-完结）_ai时代 python量化交易新模式
- 2nginx报错-bash:nginx:command not found解决方法_-bash: nginx: command not found
- 3sed 实现文件内容替换_sed替换文件中的部分内容
- 4解决Android Studio新建或导入工程总是重新下载gradle，以至于新建初始化build缓慢的问题_androidstudio新建项目重新下载
- 5bashrc文件中环境变量配置错误，导致linux命令无法正常使用的解决方案_export path=/bin:/usr/local/sbin:/usr/local/bin:/s
- 6【[ChatGPT] 从 GPT-3.5 到 GPT-5 的进阶历程 |带你全面详细的了解与学习】_gpt5和gpt3.5
- 7Java使用POI解析带图片的excel，简洁好用_easyexcel 读取单元格中的图片
- 8DDIM原理及代码(Denoising diffusion implicit models)
- 9zookeeper未授权访问（CVE-2014-0085）漏洞修复建议_zookeeper未授权访问漏洞修复
- 10vue3 全局配置Axios实例_axios在vue中全局实用
当前位置: article > 正文
改进yolov7时出现CUDA out of memory问题_yolo不断训练时显存爆了
作者：很楠不爱3 | 2024-03-18 14:10:37
赞
踩
yolo不断训练时显存爆了
1.改batch-size，可以缩小为8，4，2等。
如果还不行
2.改workers，改为4，2，1，等。
如果还不行
3.在common中改小head的数量，可以改为2.
如果还不行
4.改img-size，可以将img-size改为【320，320】
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/很楠不爱3/article/detail/263600
推荐阅读
相关标签
