从训练数据运行pianoroll_rnn_nade_generate失败,在检查点中找不到密钥解码器/ multi_rnn_cell / cell_2 / basic_lstm_cell / kernel

时间:2017-09-15 21:54:41

标签: tensorflow tensorflow-gpu magenta

我按照此处列出的教程:https://github.com/tensorflow/magenta/tree/master/magenta/models/pianoroll_rnn_nade

我成功创建了NoteSequencesSequenceExamples,对模型进行了培训并进行了评估。

当我到达Generate Pianoroll Tracks部分时,应用程序退出说明:

  

tensorflow.python.framework.errors_impl.NotFoundError:检查点中找不到密钥解码器/ multi_rnn_cell / cell_2 / basic_lstm_cell / kernel

我在一个讨论TF(https://github.com/tensorflow/tensorflow/issues/8447)中的rnn重构的线程中追踪到了这个错误(或者认为这是正确的结论)。他们提到你是否在检查点重命名变量它会起作用但在检查我的形状后我似乎没有任何遗留变量(reader.get_variable_to_shape_map()的输出):

{
'decoder/multi_rnn_cell/cell_0/basic_lstm_cell/kernel/Adam_1': [216, 512],
'decoder/multi_rnn_cell/cell_0/basic_lstm_cell/kernel/Adam': [216, 512],
'decoder/dense/bias/Adam_1': [216],
'global_step': [],
'nade/w_enc/Adam': [88, 1, 128],
'nade/w_enc': [88, 1, 128],
'decoder/dense/kernel/Adam_1': [128, 216],
'nade/w_dec_t/Adam_1': [88, 128, 1],
'nade/w_enc/Adam_1': [88, 1, 128],
'decoder/multi_rnn_cell/cell_0/basic_lstm_cell/bias': [512],
'decoder/dense/bias': [216],
'decoder/dense/bias/Adam': [216],
'beta2_power': [],
'decoder/multi_rnn_cell/cell_0/basic_lstm_cell/bias/Adam_1': [512],
'nade/w_dec_t/Adam': [88, 128, 1],
'decoder/multi_rnn_cell/cell_0/basic_lstm_cell/bias/Adam': [512],
'nade/w_dec_t': [88, 128, 1],
'beta1_power': [], 'decoder/multi_rnn_cell/cell_0/basic_lstm_cell/kernel': [216, 512],
'decoder/dense/kernel': [128, 216],
'decoder/dense/kernel/Adam': [128, 216]
}

有没有人知道我错过了什么?有没有人成功运行pianoroll_rnn_nade_generate模型?

0 个答案:

没有答案