tensorflow清除变量和图

HackerTom

于 2021-11-22 21:49:25 发布

阅读量1.7k

点赞数

分类专栏：机器学习文章标签： python tensorflow

本文链接：https://blog.csdn.net/HackerTom/article/details/121481272

版权

机器学习专栏收录该内容

121 篇文章

订阅专栏

本文探讨了如何在使用TensorFlow 1.x进行模型训练时，避免因重复定义变量导致的错误，并介绍如何在每次实验间清空计算图以实现多轮实验平均。通过实例演示了如何在`MyModel`类中正确设置和清理计算图，确保代码的正确执行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用 tensorflow 1 时，想要多次重复实验取平均，在两次实验之间需要清一次计算图，否则会报错说 xx 变量重复定义。代码形式：

# import tensorflow

class MyModel:
    def __init__(self):
        # build model

    def train(self):
        with tf.Session() as sess:
            # training


# 多 runs 取平均
for i_run in range(n_run):
    model = MyModel()
    model.train()

    # 两 runs 之间清一次计算图
    tf.reset_default_graph()

print("DONE")