亚洲乱码中文字幕综合,中国熟女仑乱hd,亚洲精品乱拍国产一区二区三区,一本大道卡一卡二卡三乱码全集资源,又粗又黄又硬又爽的免费视频

tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值

 更新時間:2020年01月04日 11:00:06   作者:Shwan_Ma  
今天小編就為大家分享一篇tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧

有時候會需要通過從保存下來的ckpt文件來觀察其保存下來的訓練完成的變量值。

ckpt文件名列表:(一般是三個文件)

xxxxx.ckpt.data-00000-of-00001

xxxxx.ckpt.index

xxxxx.ckpt.meta

import os
from tensorflow.python import pywrap_tensorflow

checkpoint_path = os.path.join("文件夾路徑", "xxxxx.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReader
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
	print("tensor_name: ", key)
	# print(reader.get_tensor(key))

以上這篇tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關文章

最新評論