Одна вещь, которую вы можете сделать, это передать --project $PROJECT_ID
в качестве параметра приложения при запуске задания ( docs ). В качестве примера на основе этого образца :
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
(или в файле, определенном в --module-name
) вы можете добавить:
args_parser.add_argument(
'--project',
help='Service Project ID where ML jobs are launched.',
required=True)
, а затем просто получить к нему доступ с помощью args.project
:
logging.info('Project ID: {}'.format(args.project))