-
Notifications
You must be signed in to change notification settings - Fork 1
/
trainingjob.json
40 lines (40 loc) · 1.04 KB
/
trainingjob.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
{
"AlgorithmSpecification":{
"TrainingImage": "072262450408.dkr.ecr.us-west-2.amazonaws.com/abalone:latest",
"TrainingInputMode": "File"
},
"RoleArn": "arn:aws:iam::072262450408:role/MLOps",
"OutputDataConfig": {
"S3OutputPath": ""
},
"ResourceConfig": {
"InstanceCount": 1,
"InstanceType": "ml.m5.xlarge",
"VolumeSizeInGB": 30
},
"TrainingJobName": "",
"HyperParameters": {
"epochs": "2000",
"layers": "2",
"dense_layer": "64",
"batch_size": "8"
},
"StoppingCondition": {
"MaxRuntimeInSeconds": 360000
},
"InputDataConfig": [
{
"ChannelName": "training",
"DataSource": {
"S3DataSource": {
"S3DataType": "S3Prefix",
"S3Uri": "",
"S3DataDistributionType": "FullyReplicated"
}
},
"ContentType": "text/csv",
"CompressionType": "None"
}
],
"Tags": []
}