tensorflow 加载模型时 报错: ValueError: No variables to save

背景

加载训练好的checkpoint

原因

在定义graph之后,当前graph没有任何变量,所以会在saver = tf.train.Saver()代码出报错.

解决办法

我们定义saver的原因,就在于加载已训练好的ckpt模型,所以可以不用定义变量。所以save应该被如下初始化

saver = tf.train.import_meta_graph('ckpt_mata_file')

你可能感兴趣的:(填坑笔记,tensorflow)