如何在不需重设/删除tensorflow中整个图的情况下重设图的权重?

时间:2019-06-20 14:06:15

标签: python tensorflow

我一直在使用tf.reset_default_graph()来测试几个图形/模型。但是,现在我想在不同的数据集上训练相同的模型/图形。因此,我想重置创建的模型的权重,而不必删除然后重新创建整个模型。请注意,我将在20个不同的数据集上训练模型。因此,与删除现有模型并为每个数据集创建一个新模型相比,仅重置权重可能是一个更简单的操作。如果我错了,请纠正我。

1 个答案:

答案 0 :(得分:1)

基于此GitHub issue,没有单个函数可以执行此操作,但是提供的解决方法是保存您的初始权重,然后使用它来重新初始化它们的权重(而不是分别随机地初始化它们)时间)。

最初,

Winit = weights.get_weights() # do this once before training

以后,

weights.set_weights(Winit)    # call each time to reset weights