Skip to content

Commit 6283c40

Browse files
awan-10nadiaya
authored andcommitted
Temporary use 1.6 TF framework version in cifar notebook so it doesn't fail. (aws#297)
1 parent caa8ce2 commit 6283c40

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

sagemaker-python-sdk/tensorflow_resnet_cifar10_with_tensorboard/tensorflow_resnet_cifar10_with_tensorboard.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +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.6',\n"
121122
" hyperparameters={'throttle_secs': 30},\n",
122123
" training_steps=1000, evaluation_steps=100,\n",
123124
" train_instance_count=2, train_instance_type='ml.c4.xlarge', \n",

0 commit comments

Comments
 (0)