读取ckpt中变量和值并赋值到新的字典中

最新推荐文章于 2019-12-03 17:12:44 发布

原创最新推荐文章于 2019-12-03 17:12:44 发布 · 574 阅读

8 ·

CC 4.0 BY-SA版权

本文介绍了一种使用Python从TensorFlow的Checkpoint文件中读取变量的方法。通过实例展示了如何获取变量名称及其对应的形状，并进一步获取每个变量的具体数值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

model_dir ="./model/"
checkpoint_path = os.path.join(model_dir, "model.ckpt-1999")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)  # tf.train.NewCheckpointReader
var_to_shape_map = reader.get_variable_to_shape_map()
print(var_to_shape_map)
d={}
for key in var_to_shape_map:
    #print("tensor_name: ", key)
    c=reader.get_tensor(key)
    #print(reader.get_tensor(key))
    d.update({key:c})