以十为单位重新初始化变量

Question

我使用一个Tensorflow tf.Saver加载一个预先训练的模型，我想通过删除（重新初始化为随机）它们的适当权重和偏差来重新训练它的一些层，然后训练这些层并保存训练的模型。我找不到重新初始化变量的方法。我试过tf.initialize_variables(fine_tune_vars)但是没有成功（我假设变量已经初始化），我也看到你可以将变量传递给tf.Saver，这样你就可以部分加载模型，但是这是我想要实现的一半（因为当我保存训练过的模型时，我希望它保存所有变量，而不仅仅是我加载的变量）。

提前谢谢你！

Answer 1

initialize_all_variables应该可以重新初始化先前初始化的变量

只是在0.10里检查了一下

tf.reset_default_graph()
a = tf.Variable(tf.ones_initializer(()))
init_op = tf.initialize_all_variables()
modify_op = a.assign(5.0)

sess = tf.InteractiveSession()
sess.run(init_op)
print(a.eval())
sess.run(modify_op)
print(a.eval())
sess.run(init_op)
print(a.eval())

结果

1.0
5.0
1.0