Tensorflow:从检查站文件中读取张量方式 - 行业资讯 - 肥雀云

在使用pre-train模型的时候,我们需从检查点文件要恢复变量。

经常出现在关卡中找不到“张量的名字找不到”。

这时候需要查看一下ckpt中到底有哪些变量

　　　　　　进口操作系统　　从tensorflow。python导入pywrap_tensorflow 　　　　checkpoint_path=os.path。加入(model_dir“model.ckpt”) 　　#从检查点文件中读取数据　　读者=pywrap_tensorflow.NewCheckpointReader (checkpoint_path) 　　var_to_shape_map=reader.get_variable_to_shape_map () 　　#打印张量名称和值　　在var_to_shape_map关键: 　　打印(“tensor_name:”键) 　　print (reader.get_tensor(关键)) 　　之前　　　　

可以显示ckpt中的张量名字和值,当然也可以用pycharm调试。

以上这篇Tensorflow:从检查站文件中读取张量方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。