解决tensorflow 释放图,删除变量问题
更新时间:2020年06月23日 14:19:56 作者:HY_JT
这篇文章主要介绍了解决tensorflow 释放图,删除变量问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
问题,在一个程序内构建好了一个图,运行完之后想重新使用这个图进行计算,或者想同时在train完的时候做test,就会提示***变量已存在。
解决办法:
待一个图运行完之后加上tf.reset_default_graph来重置默认的图。
补充知识:tensorflow 张量插入或者删除的办法
我就废话不多说了,大家还是直接看代码吧~
part1 = new_enc_in[:, 0:3 ] part2 = new_enc_in[:, 3:] ###part_new 要插入的张量 new_enc_in = tf.concat([part1, part_new, part2], 1) ##得到的张量里part_new为新插入列 ##如果要删除,则修改part2的范围 part1 = new_enc_in[:, 0:3] part2 = new_enc_in[:, 4:] new_enc_in = tf.concat([part1, part2], 1) ###原张量的第四列被删除
以上这篇解决tensorflow 释放图,删除变量问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
解读torch.cuda.amp自动混合精度训练之节省显存并加快推理速度
这篇文章主要介绍了torch.cuda.amp自动混合精度训练之节省显存并加快推理速度问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-08-08浅谈keras2 predict和fit_generator的坑
这篇文章主要介绍了浅谈keras2 predict和fit_generator的坑,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
最新评论