美文网首页
TensorFlow从checkpoint中读取模型的参数

TensorFlow从checkpoint中读取模型的参数

作者: 436宿舍 | 来源:发表于2018-08-10 14:15 被阅读0次

    from tensorflow.python.toolsimport inspect_checkpointas chkp

    chkp.print_tensors_in_checkpoint_file("../linear/model_name",tensor_name='',all_tensors=True)

    其中,模型的名字会默认保存在checkpoint中,默认保存最近的5个检查点对应的模型;

    参考:

    https://my.oschina.net/u/2362565/blog/1802226

    https://www.jianshu.com/p/60c3b084fe44

    https://www.tensorflow.org/get_started/checkpoints?hl=zh-cn

    https://tensorflow.juejin.im/get_started/checkpoints.html

    相关文章

      网友评论

          本文标题:TensorFlow从checkpoint中读取模型的参数

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