如何保存和恢复Tensorflow元素(如LSTMCell或layers.dense())?

时间:2018-09-10 22:26:59

标签: python tensorflow save

在训练了一个更复杂的模型之后,我宁愿只保存其中的一些元素。当然我可以使用tensorflow变量做到这一点。但是,对我来说,有两个特定的图形元素是一个很大的问题:LSTMCell(如果有任何区别,则为LSTMBlockCell)和密集层(tf.layers.dense())。

.ts(x?)

我特别不想保存整个模型,而只是保存并在以后还原这些特定元素。由于某些原因。

在tf.train.Saver()的var_list参数中将它们作为列表的元素传递会导致引发错误(这是有道理的,也不是变量)。 (如果有人感兴趣,我可以添加错误消息。)

如何继续保存和还原这些特定元素?有两个问题:  -我更喜欢使用tf.train.Saver(),但是如果其他方法更简单,我想知道它们  -恢复完两个元素后,我想对它们进行引用

似乎有人应该曾经问过这个问题,但是我现在已经搜索了将近两个小时,但仍未找到任何内容,因此我还是决定发布此问题。

0 个答案:

没有答案