列出检查点中变量的检查点键和形状。
用法
tf.train.list_variables(
ckpt_dir_or_file
)参数
-
ckpt_dir_or_file带有检查点文件或检查点路径的目录。
返回
-
元组列表
(key, shape)。
检查点键是检查点图中的路径。
示例用法：
import tensorflow as tf
import os
ckpt_directory = "/tmp/training_checkpoints/ckpt"
ckpt = tf.train.Checkpoint(optimizer=optimizer, model=model)
manager = tf.train.CheckpointManager(ckpt, ckpt_directory, max_to_keep=3)
train_and_checkpoint(model, manager)
tf.train.list_variables(manager.latest_checkpoint)相关用法
- Python tf.train.Coordinator.stop_on_exception用法及代码示例
- Python tf.train.ExponentialMovingAverage用法及代码示例
- Python tf.train.Checkpoint.restore用法及代码示例
- Python tf.train.Checkpoint.read用法及代码示例
- Python tf.train.CheckpointOptions用法及代码示例
- Python tf.train.ClusterSpec用法及代码示例
- Python tf.train.Checkpoint.save用法及代码示例
- Python tf.train.Checkpoint.write用法及代码示例
- Python tf.train.Coordinator用法及代码示例
- Python tf.train.CheckpointManager用法及代码示例
- Python tf.train.Checkpoint用法及代码示例
- Python tf.transpose用法及代码示例
注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.train.list_variables。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。