|
7 | 7 | "# Distributed training with TensorFlow Distribute Strategy API on Amazon SageMaker" |
8 | 8 | ] |
9 | 9 | }, |
| 10 | + { |
| 11 | + "cell_type": "markdown", |
| 12 | + "metadata": {}, |
| 13 | + "source": [ |
| 14 | + "---\n", |
| 15 | + "\n", |
| 16 | + "This notebook's CI test result for us-west-2 is as follows. CI test results in other regions can be found at the end of the notebook.\n", |
| 17 | + "\n", |
| 18 | + "\n", |
| 19 | + "\n", |
| 20 | + "---" |
| 21 | + ] |
| 22 | + }, |
10 | 23 | { |
11 | 24 | "cell_type": "markdown", |
12 | 25 | "metadata": {}, |
|
118 | 131 | "\n", |
119 | 132 | "if local_mode:\n", |
120 | 133 | " instance_type = \"local_gpu\"\n", |
121 | | - " instance_count=1\n", |
| 134 | + " instance_count = 1\n", |
122 | 135 | "else:\n", |
123 | 136 | " instance_type = \"ml.g5.xlarge\"\n", |
124 | | - " instance_count=1\n", |
| 137 | + " instance_count = 1\n", |
125 | 138 | "\n", |
126 | 139 | "mnist_estimator = TensorFlow(\n", |
127 | 140 | " entry_point=\"mnist.py\",\n", |
128 | 141 | " role=role,\n", |
129 | 142 | " instance_count=instance_count,\n", |
130 | 143 | " instance_type=instance_type,\n", |
131 | 144 | " framework_version=\"2.14.1\",\n", |
132 | | - " py_version=\"py310\"\n", |
| 145 | + " py_version=\"py310\",\n", |
133 | 146 | ")" |
134 | 147 | ] |
135 | 148 | }, |
|
280 | 293 | "\n", |
281 | 294 | "if local_mode:\n", |
282 | 295 | " instance_type = \"local_gpu\"\n", |
283 | | - " instance_count=1\n", |
| 296 | + " instance_count = 1\n", |
284 | 297 | "else:\n", |
285 | 298 | " instance_type = \"ml.g5.24xlarge\"\n", |
286 | | - " instance_count=2\n", |
| 299 | + " instance_count = 2\n", |
287 | 300 | "\n", |
288 | 301 | "mnist_estimator_distibuted = TensorFlow(\n", |
289 | 302 | " entry_point=\"mnist-distributed.py\",\n", |
|
292 | 305 | " instance_type=instance_type,\n", |
293 | 306 | " framework_version=\"2.14.1\",\n", |
294 | 307 | " py_version=\"py310\",\n", |
295 | | - " distribution={\n", |
296 | | - " \"multi_worker_mirrored_strategy\": {\n", |
297 | | - " \"enabled\": True\n", |
298 | | - " }\n", |
299 | | - " }\n", |
| 308 | + " distribution={\"multi_worker_mirrored_strategy\": {\"enabled\": True}},\n", |
300 | 309 | ")" |
301 | 310 | ] |
302 | 311 | }, |
|
317 | 326 | "source": [ |
318 | 327 | "mnist_estimator_distibuted.fit(training_data_uri)" |
319 | 328 | ] |
| 329 | + }, |
| 330 | + { |
| 331 | + "cell_type": "markdown", |
| 332 | + "metadata": { |
| 333 | + "tags": [] |
| 334 | + }, |
| 335 | + "source": [ |
| 336 | + "## Notebook CI Test Results\n", |
| 337 | + "\n", |
| 338 | + "This notebook was tested in multiple regions. The test results are as follows, except for us-west-2 which is shown at the top of the notebook.\n", |
| 339 | + "\n", |
| 340 | + "\n", |
| 341 | + "\n", |
| 342 | + "\n", |
| 343 | + "\n", |
| 344 | + "\n", |
| 345 | + "\n", |
| 346 | + "\n", |
| 347 | + "\n", |
| 348 | + "\n", |
| 349 | + "\n", |
| 350 | + "\n", |
| 351 | + "\n", |
| 352 | + "\n", |
| 353 | + "\n", |
| 354 | + "\n", |
| 355 | + "\n", |
| 356 | + "\n", |
| 357 | + "\n", |
| 358 | + "\n", |
| 359 | + "\n", |
| 360 | + "\n", |
| 361 | + "\n", |
| 362 | + "\n", |
| 363 | + "\n", |
| 364 | + "\n", |
| 365 | + "\n", |
| 366 | + "\n", |
| 367 | + "\n", |
| 368 | + "\n", |
| 369 | + "\n" |
| 370 | + ] |
320 | 371 | } |
321 | 372 | ], |
322 | 373 | "metadata": { |
|
0 commit comments