读取tf的模型

作者: 今天多云很多云 | 来源:发表于2018-06-28 13:59 被阅读0次

读取

读取保存的模型的方法

保存的模型文件一般有:

xxx.meta:通过import_meta_graph导入meta

xxx.index

xxx.data-00-of-01:这个权重都在这。通过restore路径为xxx就能读。(注意是前缀!可以用tf.train.latest_checkpoint(checkpoint_dir)这个方法获取)

注:模型的y中有用到placeholder,在sess.run()的时候肯定要feed对应的数据,因此还要根据具体placeholder的名字,从graph中使用get_operation_by_name方法获取


写入

一个训练并写入模型的例子

源码:git@github.com:hulumiaomiao/jianshu.git

相关文章

网友评论

    本文标题:读取tf的模型

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