在 Uber Ludwig 工具箱中,从头开始的训练产生了以下 10 个文件:
- 检查点
- model_hyperparameters.json
- model_weights.data-00000-of-00001
- model_weights.index
- model_weights.meta
- model_weights_progress.data-00000-of-00001
- model_weights_progress.index
- model_weights_progress.meta
- train_set_metadata.json
- training_progress.p
我想将它们(或仅需要的)导出到一个文件 saved_model.pb 中,以便能够将其加载到我的 ML Engine Google 平台。
我尝试了 freeze_graph.py 脚本,但它需要一个我没有的 .pb 文件作为输入。 (我只有上述10个文件)。如何继续获取这个 saved_model.pb 文件?我看到了几个线程,但没有一个适合我。
最佳答案
Ludwig 添加了对 SavedModel 的支持,这是 AI Platform(Cloud ML 引擎)支持的唯一模型格式
看看这个issue
关于tensorflow - 优步 Ludwig 工具箱 : export generated model to Google platform ML Engine,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54783521/