2

我正在使用谷歌云平台提交培训工作

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中的代码。从打包培训应用程序指南中，我看到您可以通过添加将参数传递给培训作业

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

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

4

2 回答 2

2

请看一下这个新的回购。

你有which 从命令task.py中读取参数gcloud

它将输入传递给该model.py文件中存在的输入。

于 2019-07-22T21:50:54.083 回答
1

我知道我可能错误地使用了我的 dockerfile，但是为了接受用户参数，我的 gcloud 查询如下所示：

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

Dockerfile

...
WORKDIR /app
COPY . /app

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

应用程序.py

def get_args():
  parser = argparse.ArgumentParser()
  parser.add_argument(
      '--user_first_arg',
      default=0)
  args = parser.parse_args()
  return args

def main():
  args = get_args()
  print(args.user_first_arg)
...
于 2020-10-01T12:55:52.380 回答