gpt4 book ai didi

python - tensorflow : how to get a list of checkpoints

转载 作者:太空宇宙 更新时间:2023-11-04 09:55:15 25 4
gpt4 key购买 nike

我需要以编程方式获取不依赖于目录列表和文件扩展验证的现有检查点列表,如果您键入:

tf.train.get_checkpoint_state('checkpoints')

您可以看到已打印此列表,但我找不到一种方法来捕获此已打印列表以便稍后对其进行迭代,有什么方法可以做到这一点吗?

最佳答案

简单地做:

tf.train.get_checkpoint_state('checkpoints').all_model_checkpoint_paths

这将为您提供检查点路径列表。 CheckpointState protobuf 定义在 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/training/checkpoint_state.proto

关于python - tensorflow : how to get a list of checkpoints,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46147593/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com