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:
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')
You can’t perform that action at this time.
0 commit comments