我不能使用一次保存的LSTM模型

时间:2019-09-11 13:24:21

标签: tensorflow kaldi

我无法在LSTM中重现训练的模型。

我想更改以下代码:

https://github.com/kaldi-asr/kaldi/blob/master/egs/wsj/s5/steps/tfrnnlm/lstm_fast.py

具体来说,我想加载Train模型并 要在保存后在另一个代码中使用它,因为 分开保存代码和加载代码是很有用的。

我保存了Train(和有效)模型,并在另一个代码中, 我挑战加载 但我遇到下一个错误;

initializer = tf.random_uniform_initializer(-config.init_scale,
                                                config.init_scale)
with tf.Session() as sess:
    saver = tf.train.import_meta_graph("~~(mypath)~~/rnnlm/model.ckpt-0.meta")
    saver.restore(sess,tf.train.latest_checkpoint("~~(my path)~~/rnnlm/."))
    valid_input = RnnlmInput(config=config, data=valid_data,name="ValidInput")
    mvalid = RnnlmModel(is_training=False, config=config, input_=valid_input)
    valid_perplexity = run_epoch(sess,mvalid)

FailedPreconditionError :
Attempting to use unitialized value embedding.

0 个答案:

没有答案