Skip to content

Commit 41891c9

Browse files
authored
Merge pull request aws#382 from yangaws/tf1.9
Update tensorflow notebooks to use sagemaker tensorflow container 1.9
2 parents d1aff33 + 861d834 commit 41891c9

File tree

9 files changed

+9
-2
lines changed

9 files changed

+9
-2
lines changed

sagemaker-python-sdk/tensorflow_abalone_age_predictor_using_keras/tensorflow_abalone_age_predictor_using_keras.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,7 @@
461461
"\n",
462462
"abalone_estimator = TensorFlow(entry_point='abalone.py',\n",
463463
" role=role,\n",
464+
" framework_version='1.9',\n",
464465
" training_steps= 100, \n",
465466
" evaluation_steps= 100,\n",
466467
" hyperparameters={'learning_rate': 0.001},\n",

sagemaker-python-sdk/tensorflow_abalone_age_predictor_using_layers/tensorflow_abalone_age_predictor_using_layers.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -506,6 +506,7 @@
506506
"\n",
507507
"abalone_estimator = TensorFlow(entry_point='abalone.py',\n",
508508
" role=role,\n",
509+
" framework_version='1.9',\n",
509510
" training_steps= 100, \n",
510511
" evaluation_steps= 100,\n",
511512
" hyperparameters={'learning_rate': 0.001},\n",

sagemaker-python-sdk/tensorflow_distributed_mnist/tensorflow_batch_transform_mnist.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@
117117
"\n",
118118
"mnist_estimator = TensorFlow(entry_point='mnist.py',\n",
119119
" role=role,\n",
120+
" framework_version='1.9',\n",
120121
" training_steps=1000, \n",
121122
" evaluation_steps=100,\n",
122123
" train_instance_count=2,\n",

sagemaker-python-sdk/tensorflow_distributed_mnist/tensorflow_distributed_mnist.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,7 @@
149149
"\n",
150150
"mnist_estimator = TensorFlow(entry_point='mnist.py',\n",
151151
" role=role,\n",
152+
" framework_version='1.9',\n",
152153
" training_steps=1000, \n",
153154
" evaluation_steps=100,\n",
154155
" train_instance_count=2,\n",

sagemaker-python-sdk/tensorflow_distributed_mnist/tensorflow_local_mode_mnist.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,7 @@
165165
"\n",
166166
"mnist_estimator = TensorFlow(entry_point='mnist.py',\n",
167167
" role=role,\n",
168+
" framework_version='1.9',\n",
168169
" training_steps=10, \n",
169170
" evaluation_steps=10,\n",
170171
" train_instance_count=2,\n",

sagemaker-python-sdk/tensorflow_iris_dnn_classifier_using_estimators/tensorflow_iris_dnn_classifier_using_estimators.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,7 @@
300300
"\n",
301301
"iris_estimator = TensorFlow(entry_point='iris_dnn_classifier.py',\n",
302302
" role=role,\n",
303-
" framework_version='1.8',\n",
303+
" framework_version='1.9',\n",
304304
" output_path=model_artifacts_location,\n",
305305
" code_location=custom_code_upload_location,\n",
306306
" train_instance_count=1,\n",

sagemaker-python-sdk/tensorflow_keras_cifar10/tensorflow_keras_CIFAR10.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,7 @@
216216
"\n",
217217
"estimator = TensorFlow(entry_point='cifar10_cnn.py',\n",
218218
" role=role,\n",
219+
" framework_version='1.9',\n",
219220
" hyperparameters={'learning_rate': 1e-4, 'decay':1e-6},\n",
220221
" training_steps=1000, evaluation_steps=100,\n",
221222
" train_instance_count=1, train_instance_type='ml.c4.xlarge')\n",

sagemaker-python-sdk/tensorflow_pipemode_example/tensorflow_pipemode_example.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@
132132
"\n",
133133
"tensorflow = TensorFlow(entry_point='pipemode.py',\n",
134134
" role=role,\n",
135+
" framework_version='1.9',\n",
135136
" input_mode='Pipe',\n",
136137
" output_path=model_artifacts_location,\n",
137138
" code_location=custom_code_upload_location,\n",

sagemaker-python-sdk/tensorflow_resnet_cifar10_with_tensorboard/tensorflow_resnet_cifar10_with_tensorboard.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@
118118
"estimator = TensorFlow(entry_point='resnet_cifar_10.py',\n",
119119
" source_dir=source_dir,\n",
120120
" role=role,\n",
121-
" framework_version='1.8',\n",
121+
" framework_version='1.9',\n",
122122
" hyperparameters={'throttle_secs': 30},\n",
123123
" training_steps=1000, evaluation_steps=100,\n",
124124
" train_instance_count=2, train_instance_type='ml.c4.xlarge', \n",

0 commit comments

Comments
 (0)