Merge pull request aws#360 from awslabs/freeze-mxnet-versions

Freeze MXNet versions

Author: djarpin

sagemaker-python-sdk/mxnet_gluon_cifar10/mxnet_cifar10_local_mode.ipynb

@@ -126,13 +126,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "m = MXNet('cifar10.py', \n",
+    "m = MXNet('cifar10.py',\n",
     "          role=role, \n",
-    "          train_instance_count=1, \n",
+    "          train_instance_count=1,\n",
     "          train_instance_type='local_gpu',\n",
-    "          hyperparameters={'batch_size': 1024, \n",
-    "                           'epochs': 50, \n",
-    "                           'learning_rate': 0.1, \n",
+    "          framework_version='1.1.0',\n",
+    "          hyperparameters={'batch_size': 1024,\n",
+    "                           'epochs': 50,\n",
+    "                           'learning_rate': 0.1,\n",
     "                           'momentum': 0.9})"
    ]
   },

sagemaker-python-sdk/mxnet_gluon_cifar10/mxnet_cifar10_with_gluon.ipynb

@@ -111,6 +111,7 @@
     "          role=role, \n",
     "          train_instance_count=2, \n",
     "          train_instance_type=\"ml.p2.xlarge\",\n",
+    "          framework_version=\"1.2.1\",\n",
     "          hyperparameters={'batch_size': 128, \n",
     "                           'epochs': 50, \n",
     "                           'learning_rate': 0.1, \n",

sagemaker-python-sdk/mxnet_gluon_mnist/mxnet_mnist_with_gluon.ipynb

@@ -98,13 +98,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "m = MXNet(\"mnist.py\", \n",
-    "          role=role, \n",
-    "          train_instance_count=1, \n",
+    "m = MXNet(\"mnist.py\",\n",
+    "          role=role,\n",
+    "          train_instance_count=1,\n",
     "          train_instance_type=\"ml.c4.xlarge\",\n",
-    "          hyperparameters={'batch_size': 100, \n",
-    "                         'epochs': 20, \n",
-    "                         'learning_rate': 0.1, \n",
+    "          framework_version=\"1.2.1\",\n",
+    "          hyperparameters={'batch_size': 100,\n",
+    "                         'epochs': 20,\n",
+    "                         'learning_rate': 0.1,\n",
     "                         'momentum': 0.9, \n",
     "                         'log_interval': 100})"
    ]

sagemaker-python-sdk/mxnet_gluon_mnist/mxnet_mnist_with_gluon_local_mode.ipynb

@@ -130,14 +130,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "m = MXNet(\"mnist.py\", \n",
-    "          role=role, \n",
-    "          train_instance_count=1, \n",
+    "m = MXNet(\"mnist.py\",\n",
+    "          role=role,\n",
+    "          train_instance_count=1,\n",
     "          train_instance_type=instance_type,\n",
-    "          hyperparameters={'batch_size': 100, \n",
-    "                         'epochs': 20, \n",
-    "                         'learning_rate': 0.1, \n",
-    "                         'momentum': 0.9, \n",
+    "          framework_version=\"1.1.0\",\n",
+    "          hyperparameters={'batch_size': 100,\n",
+    "                         'epochs': 20,\n",
+    "                         'learning_rate': 0.1,\n",
+    "                         'momentum': 0.9,\n",
     "                         'log_interval': 100})"
    ]
   },

sagemaker-python-sdk/mxnet_gluon_sentiment/mxnet_sentiment_analysis_with_gluon.ipynb

@@ -115,13 +115,14 @@
    },
    "outputs": [],
    "source": [
-    "m = MXNet(\"sentiment.py\", \n",
-    "          role=role, \n",
-    "          train_instance_count=1, \n",
+    "m = MXNet(\"sentiment.py\",\n",
+    "          role=role,\n",
+    "          train_instance_count=1,\n",
     "          train_instance_type=\"ml.c4.2xlarge\",\n",
-    "          hyperparameters={'batch_size': 8, \n",
-    "                         'epochs': 2, \n",
-    "                         'learning_rate': 0.01, \n",
+    "          framework_version=\"1.2.1\",\n",
+    "          hyperparameters={'batch_size': 8,\n",
+    "                         'epochs': 2,\n",
+    "                         'learning_rate': 0.01,\n",
     "                         'embedding_size': 50, \n",
     "                         'log_interval': 1000})"
    ]

sagemaker-python-sdk/mxnet_mnist/mxnet_mnist.ipynb

@@ -85,8 +85,9 @@
     "                        role=role,\n",
     "                        output_path=model_artifacts_location,\n",
     "                        code_location=custom_code_upload_location,\n",
-    "                        train_instance_count=1, \n",
+    "                        train_instance_count=1,\n",
     "                        train_instance_type='ml.m4.xlarge',\n",
+    "                        framework_version='1.2.1',\n",
     "                        hyperparameters={'learning_rate': 0.1})"
    ]
   },

sagemaker-python-sdk/mxnet_mnist/mxnet_mnist_with_batch_transform.ipynb

@@ -90,8 +90,9 @@
     "                        role=role,\n",
     "                        output_path=model_artifacts_location,\n",
     "                        code_location=custom_code_upload_location,\n",
-    "                        train_instance_count=1, \n",
+    "                        train_instance_count=1,\n",
     "                        train_instance_type='ml.m4.xlarge',\n",
+    "                        framework_version='1.2.1',\n",
     "                        hyperparameters={'learning_rate': 0.1})"
    ]
   },