import tensorflow as tf
# checkpoint为模型保存地址
checkpoint_path = './model'
with tf.Session() as sess:
# 得到checkpoint文件中所有的参数(名字,形状)元组
for var_name, _ in tf.contrib.framework.list_variables(checkpoint_path):
# 得到上述参数的值
var = tf.contrib.framework.load_variable(checkpoint_path, var_name)
# var_name为变量的name scope
# var 是该name scope对应的值
print(var_name,var)
输出TensorFlow 模型的变量名和对应值
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
AI从入门到入门之手写数字识别模型java方式Dense全连接神经网络实现
原創
2024-04-19 23:17:21
如何基于香橙派AIpro对视频/图像数据进行预处理
原創
2024-04-19 10:32:52
基于香橙派AIpro将开源框架模型转换为升腾模型
原創
2024-04-12 10:32:18
大模型分布式训练并行技术
原創
2024-04-10 23:29:41
TorchAcc:基于 TorchXLA 的分布式训练框架
原創
2024-04-01 23:39:48
爱奇艺 CTR 场景下的 GPU 推理性能优化
01 背景介紹 GPU 目前大量應用在了愛奇藝深度學習平臺上。GPU 擁有成百上千個處理核心,能夠並行的執行大量指令,非常適合用來做深度學習相關的計算。在 CV(計
原創
2024-03-15 15:57:51
京东广告算法架构体系建设--高性能计算方案最佳实践
原創
2024-03-15 13:45:05
语言大模型的浮点运算分配优化策略
原創
2024-03-15 00:33:17
揭秘QQ的AI绘画大模型技术——QQGC深度解析
原創
2024-03-08 00:41:36
阿里云容器服务助力万兴科技 AIGC 应用加速
原創
2023-12-28 01:18:57
轻松搭建基于服务网格的 AI 应用,然后开始玩
原創
2023-12-01 12:48:04
便捷、快速、稳定、高性能!以 GPU 实例演示 Alibaba Cloud Linux 3 对 AI 生态的支持
原創
2023-09-09 00:19:13
Deep Reinforcement Learning制作游戏AI初探
參考github案例,通過 Deep Reinforcement Learning(DQN算法)訓練能玩貪喫蛇的AI。 環境搭建 基於python + tenso
原創
2023-03-27 00:45:40
解读最佳实践:倚天710 ARM芯片的 Python+AI 算力优化
原創
2022-12-29 11:56:33