gpt4 book ai didi

tensorflow - 如何知道标签名称是tensorflow保存的模型?

转载 作者:行者123 更新时间:2023-12-02 16:20:45 24 4
gpt4 key购买 nike

如何知道tensorflow中的标签名称是“vgg16”。有什么方法可以从保存的模型图/变量转储中取回它吗?

tf.saved_model.loader.load(sess, [“vgg16”], “./data/vgg/”)

最佳答案

您可以使用 saved_model_cli 命令行工具将模型中存储的所有元数据转储到磁盘上。当您 pip install 时,该工具已包含在最新版本的 TensorFlow(当然是 1.4+)中。

例如:

saved_model_cli show --dir ./data/vgg --all

希望有帮助。

关于tensorflow - 如何知道标签名称是tensorflow保存的模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48604972/

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