如果从存储的模型加载,则无法重用LSTM单元

时间:2019-07-25 12:45:38

标签: tensorflow lstm

我有一个由保护者存储的简单LSTM模型。现在,如果我加载图形并创建具有相同名称的作用域以重用LSTM单元格,则会出现以下错误:

  

ValueError:变量Recurrent_Layers / rnn / multi_rnn_cell / cell_0 / basic_lstm_cell /内核不存在,或者不是使用tf.get_variable()创建的。您是要在VarScope中设置复用= tf.AUTO_REUSE吗?

但是我可以在原始模型(即已保存的模型)中重用它,只是当我存储它,将其加载到另一张图并尝试重用它时失败。

这是保护程序中的错误吗?

0 个答案:

没有答案