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