成都网站建设设计

将想法与焦点和您一起共享

TensorFlow输出checkpoint中的变量名与变量值方式-创新互联

废话不多说,直接看代码吧!

创新互联专业为企业提供保德网站建设、保德做网站、保德网站设计、保德网站制作等企业网站建设、网页设计与制作、保德企业网站模板建站服务,十年保德做网站经验,不只是建网站,更提供有价值的思路和整体网络服务。
import os
from tensorflow.python import pywrap_tensorflow
model_dir="/xxxxxxxxx/model.ckpt" #checkpoint的文件位置
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(model_dir)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
  print("tensor_name: ", key) #输出变量名
  print(reader.get_tensor(key))  #输出变量值

另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。


本文名称:TensorFlow输出checkpoint中的变量名与变量值方式-创新互联
文章来源:http://chengdu.cdxwcx.cn/article/deddji.html