读取tf的模型

读取

读取保存的模型的方法

保存的模型文件一般有:

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容