Update e2e-object-classification-distributed-pytorch.ipynb

Azure · Jul 17, 2023 · 5c65052 · 5c65052
1 parent e136f0b
commit 5c65052
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb b/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb
@@ -73,7 +73,7 @@
     "    workspace_name=\"<AML_WORKSPACE_NAME>\",\n",
     "    credential=credential,\n",
     ")\n",
-    "cpu_cluster = None \n",
+    "cpu_cluster = None\n",
     "gpu_cluster = None"
    ]
   },
@@ -414,7 +414,7 @@
     "    environment=\"AzureML-pytorch-1.10-ubuntu18.04-py38-cuda11-gpu@latest\",\n",
     "    compute=\"gpu-cluster\"\n",
     "    if (gpu_cluster)\n",
-    "    else None,  # No compute needs to be passed to use serverless \n",
+    "    else None,  # No compute needs to be passed to use serverless\n",
     "    distribution={\n",
     "        \"type\": \"PyTorch\",\n",
     "        # set process count to the number of gpus on the node\n",
@@ -425,9 +425,11 @@
     "    instance_count=2,\n",
     "    display_name=\"pytorch_training_sample\",\n",
     "    description=\"training a torchvision model\",\n",
-    ") \n",
-    "if gpu_cluster == None: \n",
-    "    training_job.resources = ResourceConfiguration(instance_type=\"Standard_NC6s_v3\", instance_count=2) #resources for serverless job"
+    ")\n",
+    "if gpu_cluster == None:\n",
+    "    training_job.resources = ResourceConfiguration(\n",
+    "        instance_type=\"Standard_NC6s_v3\", instance_count=2\n",
+    "    )  # resources for serverless job"
    ]
   },
   {