Skip to content

Commit a00d5ad

Browse files
authored
Documentation fix for TensorFlow training data compression (#388)
* Fix code example in TensorFlow README for using compressed training data * Fix rst syntax
1 parent b83b4e9 commit a00d5ad

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)