Skip to content

Commit

Permalink
Freeze framework mode TF notebooks to use 1.12
Browse files Browse the repository at this point in the history
  • Loading branch information
yangaws committed Dec 20, 2018
1 parent 2ac96c9 commit 3bd9a1b
Show file tree
Hide file tree
Showing 11 changed files with 13 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,7 @@
"source": [
"estimator = TensorFlow(entry_point='mnist.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" training_steps=1000, \n",
" evaluation_steps=100,\n",
" train_instance_count=1,\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -461,7 +461,7 @@
"\n",
"abalone_estimator = TensorFlow(entry_point='abalone.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" training_steps= 100, \n",
" evaluation_steps= 100,\n",
" hyperparameters={'learning_rate': 0.001},\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -506,7 +506,7 @@
"\n",
"abalone_estimator = TensorFlow(entry_point='abalone.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" training_steps= 100, \n",
" evaluation_steps= 100,\n",
" hyperparameters={'learning_rate': 0.001},\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@
"\n",
"mnist_estimator = TensorFlow(entry_point='mnist.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" training_steps=1000, \n",
" evaluation_steps=100,\n",
" train_instance_count=2,\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -149,7 +149,7 @@
"\n",
"mnist_estimator = TensorFlow(entry_point='mnist.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" training_steps=1000, \n",
" evaluation_steps=100,\n",
" train_instance_count=2,\n",
Expand Down Expand Up @@ -218,7 +218,7 @@
" print(\"========================================\")\n",
" label = np.argmax(mnist.test.labels[i])\n",
" print(\"label is {}\".format(label))\n",
" prediction = predict_response['outputs']['classes']['int64Val'][0]\n",
" prediction = predict_response['outputs']['classes']['int64_val'][0]\n",
" print(\"prediction is {}\".format(prediction))"
]
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -165,7 +165,7 @@
"\n",
"mnist_estimator = TensorFlow(entry_point='mnist.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" training_steps=10, \n",
" evaluation_steps=10,\n",
" train_instance_count=2,\n",
Expand Down Expand Up @@ -230,7 +230,7 @@
" print(\"========================================\")\n",
" label = np.argmax(mnist.test.labels[i])\n",
" print(\"label is {}\".format(label))\n",
" prediction = predict_response['outputs']['classes']['int64Val'][0]\n",
" prediction = predict_response['outputs']['classes']['int64_val'][0]\n",
" print(\"prediction is {}\".format(prediction))"
]
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -305,7 +305,7 @@
"\n",
"iris_estimator = TensorFlow(entry_point='iris_dnn_classifier.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" output_path=model_artifacts_location,\n",
" code_location=custom_code_upload_location,\n",
" train_instance_count=1,\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@
"\n",
"estimator = TensorFlow(entry_point='cifar10_cnn.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" hyperparameters={'learning_rate': 1e-4, 'decay':1e-6},\n",
" training_steps=1000, evaluation_steps=100,\n",
" train_instance_count=1, train_instance_type='ml.c4.xlarge')\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@
"\n",
"tensorflow = TensorFlow(entry_point='pipemode.py',\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" input_mode='Pipe',\n",
" output_path=model_artifacts_location,\n",
" code_location=custom_code_upload_location,\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@
"estimator = TensorFlow(entry_point='resnet_cifar_10.py',\n",
" source_dir=source_dir,\n",
" role=role,\n",
" framework_version='1.11.0',\n",
" framework_version='1.12.0',\n",
" hyperparameters={'throttle_secs': 30},\n",
" training_steps=1000, evaluation_steps=100,\n",
" train_instance_count=2, train_instance_type='ml.c4.xlarge', \n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@
"# predictable model.\n",
"env = {'SAGEMAKER_TFS_DEFAULT_MODEL_NAME': 'mobilenet_v2_140_224'}\n",
"\n",
"model = Model(model_data=model_data, role=sagemaker_role, framework_version=1.11, env=env)\n",
"model = Model(model_data=model_data, role=sagemaker_role, framework_version=1.12, env=env)\n",
"predictor = model.deploy(initial_instance_count=1, instance_type='ml.c5.xlarge')"
]
},
Expand Down

0 comments on commit 3bd9a1b

Please sign in to comment.