Slide 24
Slide 24 text
@i7i5
実行
▪boto3
sm = boto3.client("sagemaker")
sm.create_training_job(
TrainingJobName=job_name,
HyperParameters={"data_file": data_file, "cfg_file": cfg_file},
AlgorithmSpecification={"TrainingImage": repository_uri, "TrainingInputMode":
"File"},
RoleArn=role,
InputDataConfig=[
{
"ChannelName": "train",
"DataSource": {
"S3DataSource": {
"S3DataType": "S3Prefix",
"S3Uri": input_data,
"S3DataDistributionType": "FullyReplicated",
},
},
"InputMode": "File",
}
],
OutputDataConfig={"S3OutputPath": output_path},
ResourceConfig={
"InstanceType": "ml.g4dn.8xlarge",
"InstanceCount": 1,
"VolumeSizeInGB": 10,
},
StoppingCondition={
"MaxRuntimeInSeconds": 60 * 60 * 120,
},
)