背景

加载训练好的checkpoint

原因

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

解决办法

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

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

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐