Uber Ludwig Toolbox: экспортировать сгенерированную модель в платформу Google Engine ML - PullRequest
0 голосов
/ 20 февраля 2019

В наборе инструментов 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

Я хотел бы экспортировать их (или только необходимые) в один файл save_model.pb , чтобы можно было загрузить его на мою платформу ML Engine Google.

Я пробовал скрипт freeze_graph.py, но для его входа требуется файл .pb, которого у меня нет.(У меня есть только вышеупомянутые 10 файлов).Как продолжить получать этот сохраненный_модель.pb файл?Я видел несколько потоков, но ни одна из них не работает для меня.

1 Ответ

0 голосов
/ 30 июля 2019

Людвиг добавил поддержку SavedModel, единственного модельного формата, поддерживаемого платформой AI (Cloud ML Engine)

Взгляните на эту проблему

...