在Cloud Machine Learning Engine上运行TensorFlow代码时无输出

时间:2018-05-14 09:04:16

标签: tensorflow google-cloud-ml

当我在终端中运行以下内容时:

$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.indexmy_mnist_model.ckpt.metamy_mnist_model.ckpt.data-00000-of-00001

但是它们不在output文件夹中。当我在Cloud Machine Learning Engine上运行它时,我的存储桶中的指定输出文件夹中也没有任何内容。

因此模型已成功训练,但未在任何地方保存。

我的code / gcloud命令中缺少什么?

1 个答案:

答案 0 :(得分:1)

刚想通知我自己需要在剧本中处理--job-dir。从入门手册开始,我认为它是由运行培训的gcloud命令处理的。