恢复预训练的张量流模型

时间:2018-12-20 20:48:39

标签: tensorflow deep-learning save restore

我有一些使用tensorflow保存的预训练模型。对于每个模型,我都有三个文件:

我正在Ubuntu 18.0.4上运行tensorflow 1.9.0

*.ckpt.meta, *.ckpt.index and *.ckpt.data-00000-of-00001。 我编写以下代码来还原模型:

path= 'path to the folder'
with tf.Session() as sess:
    new_saver = tf.train.import_meta_graph(path + 'mobilenet_v1_0.75_224.ckpt.meta')
    new_saver.restore(sess, path + 'mobilenet_v1_0.75_224.ckpt')

我收到此错误: KeyError:“ SSTableReaderV2”

0 个答案:

没有答案