from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
savedir='model/'
print_tensors_in_checkpoint_file(savedir+"model.ckpt-60000", None, True)


checkpoints中,包含的是路径
检查TensorFlow模型checkpoint文件
本文介绍如何使用TensorFlow的inspect_checkpoint工具打印checkpoint文件中的所有张量。通过具体代码示例,展示了如何指定模型目录和文件名,以及设置是否显示所有张量的详细信息。
from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
savedir='model/'
print_tensors_in_checkpoint_file(savedir+"model.ckpt-60000", None, True)


checkpoints中,包含的是路径
您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型