updated iterative model pruning examples to PT 1.12

aws · Sep 7, 2022 · fef2253 · fef2253
1 parent 77ffe36
commit fef2253
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_alexnet.ipynb b/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_alexnet.ipynb
@@ -372,11 +372,11 @@
     "estimator = PyTorch(\n",
     "    role=sagemaker.get_execution_role(),\n",
     "    instance_count=1,\n",
-    "    instance_type=\"ml.p2.xlarge\",\n",
+    "    instance_type=\"ml.p3.2xlarge\",\n",
     "    volume_size=400,\n",
     "    source_dir=\"src\",\n",
     "    entry_point=\"train.py\",\n",
-    "    framework_version=\"1.6\",\n",
+    "    framework_version=\"1.12\",\n",
     "    py_version=\"py3\",\n",
     "    metric_definitions=[\n",
     "        {\"Name\": \"train:loss\", \"Regex\": \"loss:(.*?)\"},\n",

diff --git a/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_resnet.ipynb b/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_resnet.ipynb
@@ -340,7 +340,7 @@
     "    volume_size=400,\n",
     "    source_dir=\"src\",\n",
     "    entry_point=\"train.py\",\n",
-    "    framework_version=\"1.6\",\n",
+    "    framework_version=\"1.12\",\n",
     "    py_version=\"py3\",\n",
     "    metric_definitions=[\n",
     "        {\"Name\": \"train:loss\", \"Regex\": \"loss:(.*?)\"},\n",