查看TensorFlow检查点文件中的变量名和对应值方法 - 行业资讯

　　　　　　从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:”键) 　　打印(reader.get_tensor(关键)#删除这只是你想打印变量名　　　　

以上这篇查看TensorFlow检查点文件中的变量名和对应值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。