Gunakan estimator SageMaker AI untuk menjalankan pekerjaan pelatihan

Impor pustaka dan dependensi yang diperlukan, seperti yang ditunjukkan pada contoh kode berikut.

import boto3
import sagemaker
from sagemaker.estimator import Estimator

session = sagemaker.Session()

role = sagemaker.get_execution_role()

Berikan Uniform Resource Identifier (URI) ke image pelatihan, grup keamanan, dan subnet untuk konfigurasi VPC untuk tugas latihan Anda, seperti yang ditunjukkan pada contoh kode berikut.

image_uri = "myteam.myorg.com/docker-local/my-training-image:<IMAGE-TAG>"

security_groups = ["sg-0123456789abcdef0"]
subnets = ["subnet-0123456789abcdef0", "subnet-0123456789abcdef0"]

Untuk informasi selengkapnya tentang security_group_ids dansubnets, lihat deskripsi parameter yang sesuai di bagian Estimator SDK SageMaker Python.

Secara opsional, jika registri Docker Anda memerlukan otentikasi, Anda juga harus menentukan Nama Sumber Daya Amazon (ARN) dari AWS Lambda fungsi yang menyediakan kredensyal akses ke AI. SageMaker Contoh kode berikut menunjukkan cara menentukan ARN.

training_repository_credentials_provider_arn = "arn:aws:lambda:us-west-2:1234567890:function:test"

Untuk informasi selengkapnya tentang penggunaan gambar dalam registri Docker yang memerlukan autentikasi, lihat Menggunakan registri Docker yang memerlukan otentikasi untuk pelatihan di bawah ini.

Gunakan contoh kode dari langkah-langkah sebelumnya untuk mengkonfigurasi estimator, seperti yang ditunjukkan pada contoh kode berikut.

# The training repository access mode must be 'Vpc' for private docker registry jobs 
training_repository_access_mode = "Vpc"

# Specify the instance type, instance count you want to use
instance_type="ml.m5.xlarge"
instance_count=1

# Specify the maximum number of seconds that a model training job can run
max_run_time = 1800

# Specify the output path for the model artifacts
output_path = "s3://your-output-bucket/your-output-path"

estimator = Estimator(
    image_uri=image_uri,
    role=role,
    subnets=subnets,
    security_group_ids=security_groups,
    training_repository_access_mode=training_repository_access_mode,
    training_repository_credentials_provider_arn=training_repository_credentials_provider_arn,  # remove this line if auth is not needed
    instance_type=instance_type,
    instance_count=instance_count,
    output_path=output_path,
    max_run=max_run_time
)

Mulai pekerjaan pelatihan Anda estimator.fit dengan menelepon dengan nama pekerjaan Anda dan jalur input sebagai parameter, seperti yang ditunjukkan pada contoh kode berikut.

input_path = "s3://your-input-bucket/your-input-path"
job_name = "your-job-name"

estimator.fit(
    inputs=input_path,
    job_name=job_name
)