Halcon视觉之家-Halcon视觉技术论坛

 找回密码
 会员注册

QQ登录

只需一步,快速开始

扫一扫,微信登录

查看: 487|回复: 1

[HalconDL] 一个神经网络模型的奇怪学习行为

[复制链接]

签到天数: 2 天

连续签到: 1 天

[LV.1]初来乍到

18

主题

101

帖子

192

积分

Rank: 1

积分
192

切换助手验证会员

发表于 2022-2-11 13:07:01 | 显示全部楼层 |阅读模式

51Halcon诚邀您的加入,专注于机器视觉开发与应用技术,我们一直都在努力!

您需要 登录 才可以下载或查看,没有账号?会员注册

x
同事们,早上好!

你能告诉我如何在训练神经网络模型时减少RAM的负荷吗?

简单的例子dl_deteciion_workflow.hdev,当在GPU上训练时,它吃掉了所有的内存,进程崩溃了。显示一个内存溢出的错误。

我的电脑是英特尔i5--第十代,12个核心,16G内存,NVidia GeForce 1650 Ti。

最有趣的是,在CPU上训练时,这个问题并没有发生!!。

该用户从未签到

2

主题

14

帖子

36

积分

Rank: 1

积分
36

切换助手验证会员

发表于 2022-2-11 16:07:15 | 显示全部楼层
本帖最后由 朱颜改 于 2022-2-11 16:10 编辑

在read_dl_model之前加入这段代码试试
set_system ('cudnn_deterministic', 'true')。
您需要登录后才可以回帖 登录 | 会员注册

本版积分规则

视觉培训招生

建议您使用Chrome、Firefox、Edge、360、IE10以上等主流浏览器浏览本论坛

算子查询| 申请友链| 小黑屋| 手机版| Archiver| 业务支持|

© 2015-2021 Halcon视觉之家  X3.4  粤ICP备15095995号 粤公网安备44030602000670号

本站已安全运行:

快速回复 返回顶部 返回列表