tensorflow实现打印ckpt模型保存下的变量名称及变量的值 - 行业资讯

有时候会需要通过从保存下来的ckpt文件来观察其保存下来的训练完成的变量值。

xxxxx.ckpt.data - 00000 - 00001年的

xxxxx.ckpt.index 　　

xxxxx.ckpt.meta 　　　　　　进口操作系统　　从tensorflow。python导入pywrap_tensorflow 　　　　checkpoint_path=os.path。加入(“文件夹路径”、“xxxxx.ckpt”) 　　读者=pywrap_tensorflow.NewCheckpointReader (checkpoint_path) # tf.train.NewCheckpointReader 　　var_to_shape_map=reader.get_variable_to_shape_map () 　　在var_to_shape_map关键: 　　打印(“tensor_name:”键) 　　#打印(reader.get_tensor(关键)) 　　　　之前　　　　

以上这篇tensorflow实现打印ckpt模型保存下的变量名称及变量值就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。