Skip to content

Commit

Permalink
update (#3443)
Browse files Browse the repository at this point in the history
Co-authored-by: atqy <[email protected]>
  • Loading branch information
Zha0q1 and atqy authored Jun 7, 2022
1 parent 2199c4b commit 9065153
Showing 1 changed file with 23 additions and 23 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
"cells": [
{
"cell_type": "markdown",
"id": "28a189bb",
"id": "db45f0c5",
"metadata": {},
"source": [
"# Distributed Data Parallel EfficientNet Training with TensorFlow2 and SageMaker Distributed\n",
Expand All @@ -29,7 +29,7 @@
},
{
"cell_type": "markdown",
"id": "b70b048c",
"id": "62efb0c1",
"metadata": {},
"source": [
"## Amazon SageMaker Initialization\n",
Expand All @@ -48,7 +48,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "cb9a80f5",
"id": "94d8089f",
"metadata": {},
"outputs": [],
"source": [
Expand Down Expand Up @@ -80,7 +80,7 @@
},
{
"cell_type": "markdown",
"id": "bd2a260e",
"id": "76353ab4",
"metadata": {},
"source": [
"To verify that the role above has required permissions:\n",
Expand All @@ -94,7 +94,7 @@
},
{
"cell_type": "markdown",
"id": "a72c90b3",
"id": "9836f45e",
"metadata": {},
"source": [
"## Prepare SageMaker Training Images\n",
Expand All @@ -110,7 +110,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "aca5b15a",
"id": "fb5b7fa7",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -121,7 +121,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "2622b8f6",
"id": "ce70c7c8",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -131,7 +131,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "10d0a52d",
"id": "c712784f",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -141,7 +141,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "7ee95615",
"id": "5f00e075",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -153,7 +153,7 @@
},
{
"cell_type": "markdown",
"id": "f12251c7",
"id": "ec8919bc",
"metadata": {},
"source": [
"## Preparing FSx Input for SageMaker\n",
Expand All @@ -171,7 +171,7 @@
},
{
"cell_type": "markdown",
"id": "caf876e9",
"id": "8ace4f98",
"metadata": {},
"source": [
"## SageMaker TensorFlow Estimator function options\n",
Expand Down Expand Up @@ -202,7 +202,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "884c07bf",
"id": "d2bbed39",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -214,7 +214,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "19ade17a",
"id": "1832a42e",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -224,7 +224,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "0b831ffb",
"id": "6123f021",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -243,7 +243,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "91f2e50f",
"id": "61391941",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -270,7 +270,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "8d3312b5",
"id": "161d67cd",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -285,15 +285,15 @@
{
"cell_type": "code",
"execution_count": null,
"id": "60090fa8",
"id": "aad83355",
"metadata": {},
"outputs": [],
"source": [
"estimator = TensorFlow(\n",
" entry_point=\"main.py\",\n",
" role=role,\n",
" image_uri=docker_image,\n",
" source_dir=\"./tensorflow/efficientnet\",\n",
" source_dir=\"./SMDDP-Examples/tensorflow/efficientnet\",\n",
" instance_count=instance_count,\n",
" instance_type=instance_type,\n",
" framework_version=\"2.6\",\n",
Expand All @@ -311,7 +311,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "ee3094b5",
"id": "c9a7eafa",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -335,7 +335,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "4ff562d8",
"id": "e34fbd06",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -345,7 +345,7 @@
},
{
"cell_type": "markdown",
"id": "44c4f230",
"id": "8ccab7b1",
"metadata": {},
"source": [
"## Next steps\n",
Expand All @@ -356,7 +356,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "0438d710",
"id": "dbaa9162",
"metadata": {},
"outputs": [],
"source": [
Expand All @@ -367,7 +367,7 @@
},
{
"cell_type": "markdown",
"id": "82857412",
"id": "5963f464",
"metadata": {},
"source": [
"## Clean Up\n",
Expand Down

0 comments on commit 9065153

Please sign in to comment.