tensorflow从.meta,.data,.index格式的模型文件中restore模型

例如模型保存在checkpoint文件夹下,保存的文件名分别为:
model.index,
model.data,
model.meta

    with tf.Session(config=config) as sess:
        saver = tf.train.import_meta_graph('./checkpoint/model.meta'))
        save_path = './checkpoint/model'
        saver.restore(sess = sess, save_path = save_path)

需要注意的是saver.restore中的save_path路径在./checkpoint文件夹路径之后要写到模型的名字,不包含模型的后缀。

不然会提示错误:NotFoundError (see above for traceback): Unsuccessful TensorSliceReader constructor: Failed to find any matching files for ...

你可能感兴趣的:(tensorflow)