code-wangjun

这是tensorflow 一个经常性错误,错误的原因在于:显卡内存不够。

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

1 减少Batch 的大小

2 分析错误的位置,在哪一层出现显卡不够,比如在全连接层出现的,则降低全连接层的维度,把2048改成1042啥的

3 增加pool 层,降低整个网络的维度。

4 修改输入图片的大小

 

总结以下:想进一切办法降维,降低网络的维度。

分类:

技术点:

相关文章:

  • 2022-01-14
  • 2022-12-23
  • 2021-07-18
  • 2022-12-23
  • 2021-08-07
  • 2021-07-17
  • 2021-07-08
  • 2021-11-18
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-09-19
  • 2021-11-16
  • 2021-08-09
  • 2021-10-17
相关资源
相似解决方案