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