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

Evgeniy · 发表于 2022-2-11 13:07:01

同事们，早上好!

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

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

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

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

朱颜改 · 发表于 2022-2-11 16:07:15

本帖最后由朱颜改于 2022-2-11 16:10 编辑

在read_dl_model之前加入这段代码试试
set_system ('cudnn_deterministic', 'true')。