Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos

183 阅读 0 评论 121 点赞

我是靠谱客的博主丰富小猫咪，这篇文章主要介绍Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos，现在分享给大家，希望可以做个参考。

在CPU下跑的时候并没有报错，换成GPU后就一直OOM，后来被提醒道，显存比内存小很多啊，所以改小了batch_size,并将网络结构进行了优化，减少了词向量的维度和隐藏层节点个数。

另外，在代码里添加了GPU的相关代码：

 config = tf.ConfigProto(allow_soft_placement=True)
 config.gpu_options.allocator_type = 'BFC'
 config.gpu_options.per_process_gpu_memory_fraction = 0.80

新增的部分红线圈出：

修改完之后就不报OOM的错误了。

其他可参考：https://github.com/CharlesShang/TFFRCNN/issues/68

最后

以上就是丰富小猫咪最近收集整理的关于Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos的全部内容，更多相关Resource内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。

本文分类：Other
浏览次数：183 次浏览
发布日期：2023-10-23 01:31:36
本文链接：https://www.kaopuke.com/article/k-p-k_13_u_23_o_22_fz_13__7__14_4.html

Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos

最后

评论列表共有 0 条评论

发表评论取消回复

Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos

最后

相关文章

评论列表共有 0 条评论

发表评论 取消回复

发表评论取消回复