在Tensor Flow中保存和恢复训练有素的LSTM

Mar*_*o S 6 lstm tensorflow recurrent-neural-network

我使用BasicLSTMCell训练了LSTM分类器.如何保存模型并将其恢复以用于以后的分类?

小智 5

我们发现了同样的问题。我们不确定内部变量是否已保存。我们发现必须在创建/定义BasicLSTMCell之后创建保护程序。否则,它不会保存。


Ian*_*way 3

我自己也想知道这个问题。正如其他人指出的,在 TensorFlow 中保存模型的常用方法是使用tf.train.Saver(),但我相信这会保存 的值tf.Variables。我不确定执行此操作时是否会自动保存实现tf.Variables内部的BasicLSTMCell内容,或者是否可能需要采取另一个步骤,但如果其他所有方法都失败,则可以BasicLSTMCell轻松保存并加载到泡菜中文件。