tensorflow中内存挤爆,训练慢,可能时sess.run了图中没有构建的新节点!
tensorflow训练FPN+Faster-rcnn,每次训练到1000步左右就会cpu杀死进程,每一步训练的时间在10-15s之间,比较慢。
原因:之前在debug时,输出了图中所示的信息,这在训练中不断创建了新节点,导致graph不断增大以至于内存挤爆。在sess run中,特别是有循环的情况,要保证run的节点是图中创建好的。

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-01-10
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-01-06
猜你喜欢
  • 2021-12-14
  • 2021-09-23
  • 2021-04-29
  • 2022-12-23
  • 2021-11-26
  • 2022-12-23
  • 2021-06-20
相关资源
相似解决方案