• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

    (一)、tf.train.Saver()

    (1). tf.train.Saver() 是用来保存tensorflow训练模型的,默认保存全部参数

    (2). 用来加载参数,注:只加载存储在data中的权重和偏置项等需要训练的参数,其他一律不加载,包括meta文件中的图也不加载

    (二)、tf.train.import_meta_graph

    (1). 用来加载meta文件中的图,以及图上定义的结点参数包括权重偏置项等需要训练的参数,也包括训练过程生成的中间参数,所有参数都是通过graph调用接口get_tensor_by_name(name="训练时的参数名称")来获取

    (三)、总结

    (1). 保存使用tf.train.Saver()

    (2). 加载可以使用tf.train.import_meta_graph(".meta文件"),直接通过训练参数名称就可以获取需要的参数,但需要提前知道训练时的参数名称才能获取,要懂得tensorflow命名规则

    (3). tf.train.Saver("./checkpoints目录/")加载的缺点是只加载了训练参数,并且必须定义与之相同(shape,dtype要相同,tf.type要相同,如:我是placeholder,你也是tf.placeholder)方能使用,当你要获取训练中间参数时,需要和训练过程一样搭建相同的网络.

    补充:tf.train.import_meta_graph报KeyError

    我在模型恢复时,在执行tf.train.import_meta_graph的时候报错

    后来发现,我的模型是在服务器上训练的,服务器上tensorflow版本是1.11.0,而我在本地电脑上执行的 tf.train.import_meta_graph,我本地的tensorflow是1.5.0,我将tensorflow更新到1.11.0后,就解决了。

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

    您可能感兴趣的文章:
    • TensorFlow入门使用 tf.train.Saver()保存模型
    • Tensorflow之Saver的用法详解
    • TensorFlow Saver:保存和读取模型参数.ckpt实例
    上一篇:tensorflow中的数据类型dtype用法说明
    下一篇:教你怎么用python实现字符串转日期
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    浅谈tf.train.Saver()与tf.train.import_meta_graph的要点 浅谈,tf.train.Saver,与,tf.train.import,