Skip to content

Commit 6861e5b

Browse files
committed
Added processors for training images for consistency
1 parent b5e26fe commit 6861e5b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/sagemaker/image_uri_config/autogluon.json

+5
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@
102102
"us-west-2": "763104351884"
103103
},
104104
"repository": "autogluon-training",
105+
"processors": ["cpu", "gpu"],
105106
"py_versions": ["py38"]
106107
},
107108
"0.4.2": {
@@ -134,6 +135,7 @@
134135
"us-west-2": "763104351884"
135136
},
136137
"repository": "autogluon-training",
138+
"processors": ["cpu", "gpu"],
137139
"py_versions": ["py38"]
138140
},
139141
"0.4.3": {
@@ -166,6 +168,7 @@
166168
"us-west-2": "763104351884"
167169
},
168170
"repository": "autogluon-training",
171+
"processors": ["cpu", "gpu"],
169172
"py_versions": ["py38"]
170173
},
171174
"0.5.2": {
@@ -198,6 +201,7 @@
198201
"us-west-2": "763104351884"
199202
},
200203
"repository": "autogluon-training",
204+
"processors": ["cpu", "gpu"],
201205
"py_versions": ["py38"]
202206
},
203207
"0.6.1": {
@@ -229,6 +233,7 @@
229233
"us-west-2": "763104351884"
230234
},
231235
"repository": "autogluon-training",
236+
"processors": ["cpu", "gpu"],
232237
"py_versions": ["py38"]
233238
}
234239
}

0 commit comments

Comments
 (0)