您可以做的一件事是在启动作业(--project $PROJECT_ID
)时将docs作为应用程序参数传递。以基于this sample的示例为例:
gcloud ai-platform jobs submit training ${JOB_NAME} \
--stream-logs \
# more job configuration parameters...
--config=./config.yaml \
-- \
--project=${PROJECT_ID} \
# more application parameters...
--num-layers=3
然后,在task.py
(或在task.py
中定义的文件中,您可以添加:
--module-name
然后只需使用args_parser.add_argument(
'--project',
help='Service Project ID where ML jobs are launched.',
required=True)
进行访问:
args.project