查看模型保存的有name参数

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xky1306102chenhong/article/details/81665394

# checkpoint_dir下有四个files：meta, index, data, checkpoint
checkpoint_dir = './save/checkpoint'

reader = tf.train.NewCheckPointReader(checkpoint_dir+'/model-init')
variables = reader.get_variable_to_shape_map()

for element in variables:
	print(element)