''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with sh

这是tensorflow 一个常常性错误,错误的缘由在于:显卡内存不够。网络

解决方法就是下降显卡的使用内存,途径有如下几种措施:图片

1 减小Batch 的大小内存

2 分析错误的位置,在哪一层出现显卡不够,好比在全链接层出现的,则下降全链接层的维度,把2048改为1042啥的tensorflow

3 增长pool 层,下降整个网络的维度。方法

4 修改输入图片的大小总结

 

总结如下:想进一切办法降维,下降网络的维度。错误

相关文章
相关标签/搜索