我有一些使用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”