Исходный код SageMaker TensorFlow Estimator S3 путь загрузки

Question

Я использую оценщик SageMaker TensorFlow для обучения и указываю выходной путь для моих артефактов модели с аргументом output_path со значением s3://<bucket>/<prefix>/.

После обучения модели создается каталог с именем <training_job_name>/output в указанном output_path.

Проблема, с которой я столкнулся, заключается в том, что исходный код, который используется для обучения, по умолчанию также загружается на S3, но вместо s3://<bucket>/<prefix>/<training_job_name>/source он помещается в s3://<bucket>/<training_job_name>/source.

Так, как я могу указать путь загрузки S3 для исходного кода учебного задания, чтобы он использовал имя префикса корзины И * output_path?

Answer 1

Вы пытались использовать аргумент «code_location»: https://sagemaker.readthedocs.io/en/stable/estimators.html, чтобы указать местоположение для исходного кода?

Ниже приведен пример кода, который использует code_location

from sagemaker.tensorflow import TensorFlow

code-path = "s3://<bucket>/<prefix>"
output-path = "s3://<bucket>/<prefix>"

abalone_estimator = TensorFlow(entry_point='abalone.py',
                           role=role,
                           framework_version='1.12.0',
                           training_steps= 100, 
                           image_name=image,
                           evaluation_steps= 100,
                           hyperparameters={'learning_rate': 0.001},
                           train_instance_count=1,
                           train_instance_type='ml.c4.xlarge',
                           code_location= code-path,
                           output_path = output-path,
                           base_job_name='my-job-name'
                           )

Answer 2

Я считаю, что параметр code_location, показанный @ user3458797, является правильным ответом.

Output_path только конфигурирует местоположение S3 для сохранения результата обучения (артефакты модели и выходные файлы).

https://sagemaker.readthedocs.io/en/stable/estimators.html

Ваш сценарий обучения не будет сохранен в «output_path», если вы не переместите файл в / opt / ml / model во время обучения или не используете параметр code_location.

Пожалуйста,дайте мне знать, если есть что-то, что я могу уточнить.