网络编程
位置:首页>> 网络编程>> Python编程>> 查看TensorFlow checkpoint文件中的变量名和对应值方法

查看TensorFlow checkpoint文件中的变量名和对应值方法

作者:jingxian  发布时间:2022-07-25 08:50:17 

标签:TensorFlow,checkpoint

实例如下所示:


from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join(model_dir, "model.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
 print("tensor_name: ", key)
 print(reader.get_tensor(key)) # Remove this is you want to print only variable names

来源:https://stackoverflow.com/questions/38218174/how-to-find-the-variable-names-that-are-saved-in-a-tensorflow-checkpoint/38226516#38226516

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com