Revert "Update Pytorch Smdebug Example Notebooks With SM SDK V2" (aws#1521)

This reverts commit 0f00154.

Author: mchoi8739

sagemaker-debugger/pytorch_custom_container/pytorch_byoc_smdebug.ipynb

@@ -192,10 +192,10 @@
     "estimator = PyTorch(entry_point=smdebug_mnist_script,\n",
     "                  base_job_name='smdebugger-demo-mnist-pytorch',\n",
     "                  role=role,\n",
-    "                  instance_count=1,\n",
-    "                  instance_type='ml.m4.xlarge',\n",
-    "                  volume_size=400,\n",
-    "                  max_run=3600,\n",
+    "                  train_instance_count=1,\n",
+    "                  train_instance_type='ml.m4.xlarge',\n",
+    "                  train_volume_size=400,\n",
+    "                  train_max_run=3600,\n",
     "                  hyperparameters=hyperparameters,\n",
     "                  framework_version='1.3.1',\n",
     "                  py_version='py3',\n",

sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_alexnet.ipynb

@@ -348,9 +348,9 @@
     "from sagemaker.pytorch import PyTorch\n",
     "\n",
     "estimator = PyTorch(role=sagemaker.get_execution_role(),\n",
-    "                  instance_count=1,\n",
-    "                  instance_type='ml.p2.xlarge',\n",
-    "                  volume_size=400,\n",
+    "                  train_instance_count=1,\n",
+    "                  train_instance_type='ml.p2.xlarge',\n",
+    "                  train_volume_size=400,\n",
     "                  source_dir='src',\n",
     "                  entry_point='train.py',\n",
     "                  framework_version='1.3.1',\n",

sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_resnet.ipynb

@@ -309,9 +309,9 @@
     "from sagemaker.pytorch import PyTorch\n",
     "\n",
     "estimator = PyTorch(role=sagemaker.get_execution_role(),\n",
-    "                  instance_count=1,\n",
-    "                  instance_type='ml.p3.2xlarge',\n",
-    "                  volume_size=400,\n",
+    "                  train_instance_count=1,\n",
+    "                  train_instance_type='ml.p3.2xlarge',\n",
+    "                  train_volume_size=400,\n",
     "                  source_dir='src',\n",
     "                  entry_point='train.py',\n",
     "                  framework_version='1.3.1',\n",