Нет вывода при запуске кода TensorFlow на движке Cloud Machine Learning - PullRequest
0 голосов
/ 14 мая 2018

Когда я запускаю в терминале следующее:

$MODEL_DIR=output
gcloud ml-engine local train     --module-name trainer.task     --package-path trainer/     --job-dir $MODEL_DIR

Он работает успешно, но я ничего не получаю в выходной папке.Хотя в соответствии с этим я должен увидеть некоторые файлы и контрольные точки: https://cloud.google.com/ml-engine/docs/tensorflow/getting-started-training-prediction

В коде у меня есть эта строка для сохранения моей модели:

save_path = saver.save(sess, "./my_mnist_model.ckpt")

, которая генерирует следующие файлы вактивный каталог: my_mnist_model.ckpt.index, my_mnist_model.ckpt.meta, my_mnist_model.ckpt.data-00000-of-00001

Однако они не находятся в папке output.И когда я запускаю его на Cloud Machine Learning Engine, я не получаю ничего в указанной выходной папке в моем корзине.

Таким образом, модель успешно обучена, но нигде не сохранена.

Чего мне не хватает в моей команде code / gcloud?

1 Ответ

0 голосов
/ 14 мая 2018

Просто сам понял, что мне нужно разобраться с самим --job-dir в скрипте. Из руководства по началу работы я подумал, что оно обрабатывается командой gcloud, которая запускает обучение.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...