当我在终端中运行以下内容时:
$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命令中缺少什么?
答案 0 :(得分:1)
刚想通知我自己需要在剧本中处理--job-dir。从入门手册开始,我认为它是由运行培训的gcloud命令处理的。