Skip to content

Commit c9c9fd5

Browse files
winstonawsPiali Das
authored and
Piali Das
committed
Documentation fix for TensorFlow training data compression (aws#388)
* Fix code example in TensorFlow README for using compressed training data * Fix rst syntax
1 parent 420faa9 commit c9c9fd5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sagemaker/tensorflow/README.rst

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -819,7 +819,10 @@ If your TFRecords are compressed, you can train on Gzipped TF Records by passing
819819

820820
.. code:: python
821821
822-
tf_estimator.fit('s3://bucket/path/to/training/data', compression='Gzip')
822+
from sagemaker.session import s3_input
823+
824+
train_s3_input = s3_input('s3://bucket/path/to/training/data', compression='Gzip')
825+
tf_estimator.fit(train_s3_input)
823826
824827
825828
You can learn more about ``PipeModeDataset`` in the sagemaker-tensorflow-extensions repository: https://github.com/aws/sagemaker-tensorflow-extensions

0 commit comments

Comments
 (0)