tensorflow 保存训练模型ckpt 查看ckpt文件中的变量名和对应值
时间: 2024-03-10 21:47:37 浏览: 132
tensorflow实现将ckpt转pb文件的方法
可以使用 TensorFlow 自带的工具 `inspect_checkpoint.py` 来查看 ckpt 文件中的变量名和对应值。
具体操作步骤如下:
1. 打开命令行终端,进入到包含 ckpt 文件的目录下。
2. 输入以下命令,查看 ckpt 文件中的所有变量名:
```
python -m tensorflow.python.tools.inspect_checkpoint --file_name=model.ckpt
```
其中,`model.ckpt` 是你保存的 ckpt 文件名,根据自己的实际情况修改。
3. 输入以下命令,查看 ckpt 文件中指定变量名的值:
```
python -c "import tensorflow as tf; print(tf.train.load_variable('model.ckpt', 'variable_name'))"
```
其中,`model.ckpt` 是你保存的 ckpt 文件名,`variable_name` 是你要查看的变量名,根据自己的实际情况修改。
运行命令后,会显示出指定变量名的值。
阅读全文