Tensorflow,tf.train.Saver保存了什么?

cha*_*255 2 tensorflow

我想知道在每次训练后使用tf.train.Saver()保存模型时究竟保存了什么。与我习惯使用Keras模型的文件相比,该文件似乎有点大。现在,每次保存时,我的RNN占用900 MB。有什么办法告诉保护程序只保存可训练的参数?我还想要一种只保存部分模型的方法。我知道我可以获取我定义的变量并使用numpy格式保存它们,但是当我使用RNN类时,我无法直接访问它们的权重,我查看了代码,没有像get_weights这样的东西了。

Yar*_*tov 5

您可以提供要保存在Saver构造函数中的变量列表,即saver=tf.train.Saver(var_list=tf.trainable_variables())