Tensorflow: 从checkpoint文件中读取tensor
2017-05-08 20:10
288 查看
在使用pre-train model时候,我们需要restore variables from checkpoint files.
经常出现在checkpoint 中找不到”Tensor name not found”.
这时候需要查看一下ckpt中到底有哪些变量
可以显示ckpt中的tensor名字和值,当然也可以用pycharm调试。
经常出现在checkpoint 中找不到”Tensor name not found”.
这时候需要查看一下ckpt中到底有哪些变量
import os from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join(model_dir, "model.ckpt") # Read data from checkpoint file reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.get_variable_to_shape_map() # Print tensor name and values for key in var_to_shape_map: print("tensor_name: ", key) print(reader.get_tensor(key))
可以显示ckpt中的tensor名字和值,当然也可以用pycharm调试。
相关文章推荐
- Tensorflow: 从checkpoint文件中读取tensor
- Tensorflow: 读取ckpt文件中的tensor
- Tensorflow建立与读取TFrecorder文件
- tensorflow中的TFRecord格式文件的写入和读取
- TensorFlow基础5:TFRecords文件的存储与读取讲解及代码实现
- Tensorflow学习笔记-通过slim读取TFRecord文件
- TensorFlow学习Day3读取csv文件,动手写个logistic,softmax分类模型
- Tensorflow从文件读取数据
- Tensorflow 从bin文件中读取数据并
- Tensorflow自定义读取文件
- TensorFlow的文件保存与读取——variables_to_restore函数
- tensorflow存取,读取,及保存的文件的含义
- TensorFLow 读取图片1:初探四种从文件读取的方式
- tensorflow 批量读取csv文件用于做深度学习算法相关
- TensorFlow模型文件保存和读取
- TensorFlow - 读取CSV格式文件
- Tensorflow直接读取二进制文件
- Tensorflow直接读取CSV文件
- tensorflow之从文件中读取数据(适用场景:大规模数据集,亲测有效~)
- 如何用Tensorflow训练模型成pb文件(二)——基于tfrecord的读取