我在Tensorflow中训练了一个具有批量规范的模型.我想保存模型并将其恢复以供进一步使用.批量规范由.完成
def batch_norm(input, phase):
return tf.layers.batch_normalization(input, training=phase)
阶段True在训练False期间和测试期间.
这似乎只是在呼唤
saver = tf.train.Saver()
saver.save(sess, savedir + "ckpt")
不会很好,因为当我恢复模型时,它首先说成功恢复.它还说Attempting to use uninitialized value batch_normalization_585/beta如果我只是在图中运行一个节点.这是否与正确保存模型或我错过的其他内容有关？