reader = tf.pywrap_tensorflow.NewCheckpointReader(ckpt_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
value=reader.get_tensor(key)
if isinstance(value, np.ndarray):
print(‘value:’,np.shape(value),value)
else:
print(‘value:’,type(value),value)
tensorflow打印checkpoint里的变量名称和值
最新推荐文章于 2022-03-07 21:39:55 发布