Question

我正在使用

在Google云平台上提交培训工作

gcloud ai-platform jobs submit training $JOB_NAME
               --scale-tier basic
               --package-path $TRAINING_PACKAGE_PATH
               --module-name $MAIN_TRAINER_MODULE
               --job-dir $JOB_DIR
               --runtime-version $RUNTIME_VERSION
               --python-version $PYTHON_VERSION
               --region $REGION

我的训练代码看起来有点像在线教程cloudml-samples中的代码。在Packaging a Training Application指南中，我看到您可以通过添加

将参数传递给培训工作

-- \
--user_first_arg=first_arg_value \
--user_second_arg=second_arg_value

但是在任何地方我都能找到如何从训练代码中读取参数。有什么建议么？谢谢

Answer 1

请看看这个新的repo。

您有task.py可以从gcloud命令中读取参数

将输入传递到this文件中存在的model.py。

Answer 2

我知道我可能不正确地使用了dockerfile，但是为了接受用户args，我的gcloud查询看起来像这样：

Something went wrong

Dockerfile

gcloud ai-platform jobs submit training $JOB_NAME  /
--region $REGION   /
--master-image-uri $IMAGE_URI   /
--  /
app.py --user_first_arg=first_arg_value

app.py

...
WORKDIR /app
COPY . /app

ENTRYPOINT ["python"]
CMD ["app.py"]

Google Cloud Platform提交了培训工作，如何从培训代码中读取USER_ARGS？

2 个答案: