File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -718,7 +718,7 @@ follows:
718
718
719
719
from sagemaker.tensorflow import TensorFlow
720
720
721
- tf_estimator = TensorFlow('tf-train.py', role='SageMakerRole',
721
+ tf_estimator = TensorFlow(entry_point= 'tf-train.py', role='SageMakerRole',
722
722
training_steps=10000, evaluation_steps=100,
723
723
train_instance_count=1, train_instance_type='ml.p2.xlarge')
724
724
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
1042
1042
1043
1043
from sagemaker.tensorflow import TensorFlow
1044
1044
1045
- tf_estimator = TensorFlow('tf-train.py', role='SageMakerRole',
1045
+ tf_estimator = TensorFlow(entry_point= 'tf-train.py', role='SageMakerRole',
1046
1046
training_steps=10000, evaluation_steps=100,
1047
1047
train_instance_count=1, train_instance_type='ml.p2.xlarge')
1048
1048
tf_estimator.fit('s3://bucket/path/to/training/data')
You can’t perform that action at this time.
0 commit comments