diff --git a/cli/jobs/pipelines/cifar-10/pipeline.yml b/cli/jobs/pipelines/cifar-10/pipeline.yml
index aa3995ed98..5e62cef75d 100644
--- a/cli/jobs/pipelines/cifar-10/pipeline.yml
+++ b/cli/jobs/pipelines/cifar-10/pipeline.yml
@@ -51,9 +51,9 @@ jobs:
     compute: azureml:gpu-cluster
     distribution:
       type: pytorch
-      process_count_per_instance: 2
+      process_count_per_instance: 1
     resources:
-      instance_count: 1
+      instance_count: 2
     inputs:
       cifar: ${{parent.jobs.get_data.outputs.cifar}}
       model_dir: ${{parent.jobs.train_model.outputs.model_dir}}
diff --git a/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/eval-model.yml b/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/eval-model.yml
index 9834e9e0b5..5dc6e08248 100644
--- a/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/eval-model.yml
+++ b/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/eval-model.yml
@@ -14,7 +14,7 @@ inputs:
 code: ./eval-model
 distribution:
   type: pytorch
-  process_count_per_instance: 2
+  process_count_per_instance: 1
 resources:
   instance_count: 2
 environment: azureml:AzureML-pytorch-1.9-ubuntu18.04-py37-cuda11-gpu:6
diff --git a/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train-model.yml b/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train-model.yml
index ae5357ef3b..6745182672 100644
--- a/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train-model.yml
+++ b/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train-model.yml
@@ -17,7 +17,7 @@ outputs:
 code: ./train-model
 distribution:
   type: pytorch
-  process_count_per_instance: 2
+  process_count_per_instance: 1
 resources:
   instance_count: 2
 environment: azureml:AzureML-pytorch-1.9-ubuntu18.04-py37-cuda11-gpu:6
diff --git a/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train_cifar_10_with_pytorch.ipynb b/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train_cifar_10_with_pytorch.ipynb
index cc68cd3c79..d7ed446ae2 100644
--- a/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train_cifar_10_with_pytorch.ipynb
+++ b/sdk/python/jobs/pipelines/2b_train_cifar_10_with_pytorch/train_cifar_10_with_pytorch.ipynb
@@ -158,14 +158,12 @@
     "    train_model = train_model_func(epochs=1, cifar=get_data.outputs.cifar)\n",
     "    train_model.compute = \"gpu-cluster\"\n",
     "    train_model.outputs.model_dir.mode = \"upload\"\n",
-    "    train_model.resources.instance_count = 1\n",
     "\n",
     "    # define the job to evaluate the model\n",
     "    eval_model = eval_model_func(\n",
     "        cifar=get_data.outputs.cifar, model_dir=train_model.outputs.model_dir\n",
     "    )\n",
     "    eval_model.compute = \"gpu-cluster\"\n",
-    "    eval_model.resources.instance_count = 1\n",
     "\n",
     "\n",
     "pipeline_job = train_cifar_10_with_pytorch()\n",