Error:OOM when allocating tensor with shape[......]

问题描述: 训练模型时,遇到: 原因: 内存不够,OOM即ran out of memory 可以使用top命令动态监听运行时间非常短的程序的内存使用情况:参考 启用2个session,一个session运行top命令,另一个session运行指定的程序,top命令会输出与这个命令相关的资源使用情况。 解决方法: 最直观的方式就是减小batch_size或者hidden_layer中的单元数 减少
相关文章
相关标签/搜索