还原ComputationGraph时出现内存不足错误-dl4j

问题描述

我正在尝试在树莓派3 B +上运行一个简单的神经网络，但是该程序在加载预先训练的模型时被杀死。它说“ Out of Memory”（内存不足），所以我用-Xmx增加了堆空间。在我的Mac上，它可以工作，但是在树莓派上，错误仍然相同。有谁知道为什么它不能在覆盆子上工作？！

以下一些有用的信息： Mac：

[main] INFO org.nd4j.linalg.factory.Nd4jBackend - Loaded [cpuBackend] backend
[main] INFO org.nd4j.nativeblas.NativeOpsHolder - Number of threads used for linear algebra: 1
[main] WARN org.nd4j.linalg.cpu.nativecpu.cpundarrayFactory - *********************************** cpu Feature Check Warning ***********************************
[main] WARN org.nd4j.linalg.cpu.nativecpu.cpundarrayFactory - Warning: Initializing ND4J with Generic x86 binary on a cpu with AVX/AVX2 support
[main] WARN org.nd4j.linalg.cpu.nativecpu.cpundarrayFactory - Using ND4J with AVX/AVX2 will improve performance. See deeplearning4j.org/cpu for more details
[main] WARN org.nd4j.linalg.cpu.nativecpu.cpundarrayFactory - Or set environment variable ND4J_IGnorE_AVX=true to suppress this warning
[main] WARN org.nd4j.linalg.cpu.nativecpu.cpundarrayFactory - *************************************************************************************************
[main] INFO org.nd4j.nativeblas.Nd4jBlas - Number of threads used for OpenMP BLAS: 4
[main] INFO org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner - Backend used: [cpu]; OS: [Mac OS X]
[main] INFO org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner - Cores: [8]; Memory: [0.5GB];
[main] INFO org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner - Blas vendor: [OPENBLAS]

RaspBerry Pi：

[main] INFO org.nd4j.linalg.factory.Nd4jBackend - Loaded [cpuBackend] backend
[main] INFO org.nd4j.nativeblas.NativeOpsHolder - Number of threads used for linear algebra: 4
[main] INFO org.nd4j.nativeblas.Nd4jBlas - Number of threads used for OpenMP BLAS: 4
[main] INFO org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner - Backend used: [cpu]; OS: [Linux]
[main] INFO org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner - Cores: [4]; Memory: [0.5GB];
[main] INFO org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner - Blas vendor: [OPENBLAS]
Killed

当我检查上一个终止的进程时，它显示了以下错误消息：

Out of memory: Killed process 1802 (java) total-vm:3433160kB,anon-RSS:688420kB,file-RSS:1672kB,shmem-RSS:0kB,UID:1000 pgtables:1744kB oom_score_adj:0

如果有人有一个主意，那就太好了！谢谢！

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）