有時候會需要通過從保存下來的ckpt文件來觀察其保存下來的訓練完成的變量值。
ckpt文件名列表：（一般是三個文件）
xxxxx.ckpt.data-00000-of-00001
xxxxx.ckpt.index
xxxxx.ckpt.meta
|
1
2
3
4
5
6
7
8
9
|
import oscheckpoint_path = os.path.join("文件夾路徑", "xxxxx.ckpt")reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReadervar_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實現打印ckpt模型保存下的變量名稱及變量值就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持服務器之家。
原文鏈接：https://blog.csdn.net/shwan_ma/article/details/84941051
