Skip to content

Commit e82fb4f

Browse files
nkconnorlukmis
authored andcommitted
Update TensorFlow examples following API change (#44)
1 parent 05d4b0b commit e82fb4f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,7 @@ follows:
718718
719719
from sagemaker.tensorflow import TensorFlow
720720
721-
tf_estimator = TensorFlow('tf-train.py', role='SageMakerRole',
721+
tf_estimator = TensorFlow(entry_point='tf-train.py', role='SageMakerRole',
722722
training_steps=10000, evaluation_steps=100,
723723
train_instance_count=1, train_instance_type='ml.p2.xlarge')
724724
tf_estimator.fit('s3://bucket/path/to/training/data')
@@ -1042,7 +1042,7 @@ The following code sample shows how to train a custom TensorFlow script 'tf-trai
10421042
10431043
from sagemaker.tensorflow import TensorFlow
10441044
1045-
tf_estimator = TensorFlow('tf-train.py', role='SageMakerRole',
1045+
tf_estimator = TensorFlow(entry_point='tf-train.py', role='SageMakerRole',
10461046
training_steps=10000, evaluation_steps=100,
10471047
train_instance_count=1, train_instance_type='ml.p2.xlarge')
10481048
tf_estimator.fit('s3://bucket/path/to/training/data')

0 commit comments

Comments
 (0)