3

在 Uber Ludwig 工具箱中，从头开始的训练产生了以下 10 个文件：

  1. 检查点
  2. model_hyperparameters.json
  3. model_weights.data-00000-of-00001
  4. model_weights.index
  5. model_weights.meta
  6. model_weights_progress.data-00000-of-00001
  7. model_weights_progress.index
  8. model_weights_progress.meta
  9. train_set_metadata.json
  10. training_progress.p

我想将它们（或仅需要的）导出到一个文件saved_model.pb以便能够将其加载到我的 ML Engine Google 平台。

我尝试了 freeze_graph.py 脚本，但它需要一个我没有的 .pb 文件作为输入。（我只有上面提到的10个文件）。如何继续获取这个saved_model.pb文件？我看到了几个线程，但没有一个对我有用。

4

1 回答 1

1

Ludwig 添加了对 SavedModel 的支持，这是 AI Platform（Cloud ML Engine）支持的唯一模型格式

看看这个问题

于 2019-07-30T06:42:11.463 回答