tensorflow学习笔记

一、saved model(tf.saved_model)

a. builder = tf.saved_model.builder.SavedModelBuilder(export_dir):

创建一个保存模型的实例对象

b. tf.saved_model.utils.build_tensor_info(tensor):

构建TensorInfo protobuf,根据输入的tensor构建相应的protobuf,返回的tensorinfo中包含输入tensor的name,shape,dtype信息。

c. tf.saved_model.signature_def_utils.build_signature_def:

构建 SignatureDef protobuf,并返回 SignatureDef protobuf

d. tf.saved_model.utils.get_tensor_from_tensor_info:

根据一个 TensorInfo protobuf 解析出一个 tensor

e. builder.add_meta_graph_and_variables(sess, tags:

保存会话对象中的graph和所有变量

f. builder.save():将内建的savedModel protobuf写入磁盘

二、Saver(.ckpt文件的保存和加载)

a. saver= tf.train.Saver(max_to_keep=5 ):保存和加载模型,max_to_keep参数表示要保留的最近检查文件的最大数量,创建新文件时,将删除旧文件

b.  saver.save(s

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值