gpt4 book ai didi

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

转载 作者:qq735679552 更新时间:2022-09-29 22:32:09 31 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章从训练好的tensorflow模型中打印训练变量实例由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

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

?
1
2
3
4
import tensorflow as tf
reader = tf.train.NewCheckpointReader( 'path/alexnet/model-330000' )
dic = reader.get_variable_to_shape_map()
print dic

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

打印变量 。

?
1
2
3
4
w = reader.get_tensor( "fc1/W" )
print type (w)
print w.shape
print w[ 0 ]

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

以上这篇从训练好的tensorflow模型中打印训练变量实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我.

原文链接:https://blog.csdn.net/aiseu001/article/details/79851176 。

最后此篇关于从训练好的tensorflow模型中打印训练变量实例的文章就讲到这里了,如果你想了解更多关于从训练好的tensorflow模型中打印训练变量实例的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com