在Uber Ludwig Toolbox中,从头开始产生以下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
- triending_progress.p
我想将它们(或仅需要的)导出到一个文件 saved_model.pb 以便能够将其加载到我的ML Engine Google平台上。
我尝试了freeze_graph.py脚本,但是它需要作为我没有的输入a .pb文件。(我只有上述10个文件)。如何继续获取此 saved_model.pb 文件?我看到了几个线程,但是它们都不适合我。
ludwig增加了对SavedModel的支持
看这个问题