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://ybzwz.com/article/deddji.html