美文网首页
tensorflow加载meta模型

tensorflow加载meta模型

作者: geofer | 来源:发表于2017-12-08 14:30 被阅读0次

模型save后会产生三个文件:

model.ckpt.data-00000-of-00001
model.ckpt.index
model.ckpt.meta

这种模型的加载方法:

with tf.Session() as sess:
saver = tf.train.import_meta_graph('/tmp/model.ckpt.meta')
saver.restore(sess, "/tmp/model.ckpt")

相关文章

网友评论

      本文标题:tensorflow加载meta模型

      本文链接:https://www.haomeiwen.com/subject/bzyeixtx.html