首页 > 其他分享 >tensorflow 输出权重的值

tensorflow 输出权重的值

时间:2023-05-09 19:11:36浏览次数:42  
标签:输出 sess name 权重 get fc tf tensorflow logits

1. 确定权重名称:

tvars1 = tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)

for tmp in tvars1:

  print('all-->',tmp.name)

2. 根据网络结构从1中找到想要打印的权重名称 weight_name,通过下面的方式进行打印

fc_logits=tf.get_default_graph().get_tensor_by_name(weight_name)

with tf.Session() as sess:
  init_op = tf.group(tf.global_variables_initializer(),tf.local_variables_initializer())
  sess.run(init_op)
  fc_logits_ = sess.run(fc_logits,feed_dict={input_placeholder: gen_input,gt: label})
  print('fc_logits_:',fc_logits_)

 

  

标签:输出,sess,name,权重,get,fc,tf,tensorflow,logits
From: https://www.cnblogs.com/BlogLwc/p/17385983.html

相关文章