Tensorflow所有变量只返回权重



我有一个Tensorflow v1会话图(经过训练),并希望从中检索权重和偏差,以将它们输入到具有相同架构的Keras模型中。我已经设法得到了所有的权重与以下呼叫,但没有偏差。我试着:

tf.all_variables()
tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES)
tf.get_collection(tf.GraphKeys.BIASES) 

对于如何获得偏差有什么想法吗?

解决方案:模型最初是用卷积层训练的,只有可训练的权重,而不是偏差。为了使用Keras正确地恢复模型,将偏差设置为0。

最新更新