从训练好的tensorflow模型中打印训练变量

转载:https://blog.csdn.net/aiseu001/article/details/79851176

从tensorflow 训练后保存的模型中打印训变量:使用tf.train.NewCheckpointReader()

 
  1. import tensorflow as tf

  2. reader = tf.train.NewCheckpointReader('path/alexnet/model-330000')

  3. dic = reader.get_variable_to_shape_map()

  4. print dic

打印变量

 
  1. w = reader.get_tensor("fc1/W")

  2. print type(w)

  3. print w.shape

  4. print w[0]

猜你喜欢

转载自blog.csdn.net/m0_37870649/article/details/81776944