tensorflow 运行时内存持续增长(溢出)

tensorflow 训练部分代码几乎都采用循环,如果在循环中用到tf的op操作,就会导致节点越来越多,图的大小也不停的增大,故而内存会持续增长 常见情况有几种 1、sess.run(a+b)       写成sess.run(add_a_b)最好 2、tf.train.Saver() 3.不要使用额外的图 原因是在对每一张图片正向传播结束后,都会在留下中间信息。具体地说是在我将正向传播的代码与模
相关文章
相关标签/搜索