Как я могу получить текущее название должности в сценарии учебного задания SageMaker?

Question

Я пишу учебное задание на AWS -SageMaker framework.

Для некоторых требований ему нужно знать имя задания, на котором работает текущий.

Я знаю этот код работает для него ...

import sagemaker_containers
env = sagemaker_containers.training_env()
job_name = env['job_name']

Но пакет sagemaker_containers устарел. (Я прочитал, что на это GitHub )

Что мне делать?

Я только начал изучать эту платформу в прошлом месяце. Буду признателен за любой совет. Спасибо.

Answer 1

Для старых контейнеров, использующих устаревший sagemaker_containers, описанный вами подход является правильным.

Для более новых контейнеров, использующих sagemaker-training-toolkit, вы получаете информацию о среде таким образом : https://github.com/aws/sagemaker-training-toolkit#get -information-about-the-container-environment

from sagemaker_training import environment

env = environment.Environment()

job_name = env["job_name"]

Вы можете проверить DL C Release Notes , чтобы узнать, что установлено в каждая версия.