美文网首页
tensorflow获取训练过程中变量的值

tensorflow获取训练过程中变量的值

作者: 刘大力_ | 来源:发表于2018-04-25 17:08 被阅读0次

训练时,想要获取过程中的值,网络上能查到以下几种方式:

#第一种

sess.run(train, feed_dict=feed_dict)

print(loss.eval(feed_dict, sess))

还有一种

#第二种

_, loss_value = sess.run((train, loss)) 

print(loss_value)

两种都可以使用。另外,第二种参数如下:

run(fetches,   feed_dict=None,    options=None,    run_metadata=None)

在使用tf.sess.run时,fetches 可以传入list。如果 fetches 是一个元素就返回一个值;若 fetches 是一个 list,则返回 list 的值,若 fetches 是一个字典类型,则返回和 fetches 同 keys 的字典。

相关文章

网友评论

      本文标题:tensorflow获取训练过程中变量的值

      本文链接:https://www.haomeiwen.com/subject/nluylftx.html