【发布时间】:2018-05-14 23:39:30
【问题描述】:
当我在终端中运行以下命令时:
$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 上运行它时,我的存储桶中的指定输出文件夹中也没有任何内容。
所以模型训练成功,但没有保存在任何地方。
我的代码/gcloud 命令中缺少什么?
【问题讨论】:
标签: tensorflow google-cloud-ml