Skip to content

Commit 864c653

Browse files
authored
Fix docs about s3_input usage (#258)
1 parent 9a27c3b commit 864c653

File tree

4 files changed

+6
-5
lines changed

4 files changed

+6
-5
lines changed

src/sagemaker/estimator.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,8 @@ def fit(self, inputs, wait=True, logs=True, job_name=None):
165165
training data, you can specify a dict mapping channel names
166166
to strings or :func:`~sagemaker.session.s3_input` objects.
167167
* (sagemaker.session.s3_input) - channel configuration for S3 data sources that can provide
168-
additional information about the training dataset. See :func:`sagemaker.session.s3_input`
169-
for full details.
168+
additional information as well as the path to the training dataset.
169+
See :func:`sagemaker.session.s3_input` for full details.
170170
wait (bool): Whether the call should wait until the job completes (default: True).
171171
logs (bool): Whether to show the logs produced by the job.
172172
Only meaningful when wait is True (default: True).

src/sagemaker/session.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -890,7 +890,7 @@ def __init__(self, s3_data, distribution='FullyReplicated', compression=None,
890890
s3_data (str): Defines the location of s3 data to train on.
891891
distribution (str): Valid values: 'FullyReplicated', 'ShardedByS3Key'
892892
(default: 'FullyReplicated').
893-
compression (str): Valid values: 'Gzip', 'Bzip2', 'Lzop' (default: None).
893+
compression (str): Valid values: 'Gzip', None (default: None). This is used only in Pipe input mode.
894894
content_type (str): MIME type of the input data (default: None).
895895
record_wrapping (str): Valid values: 'RecordIO' (default: None).
896896
s3_data_type (str): Value values: 'S3Prefix', 'ManifestFile'. If 'S3Prefix', ``s3_data`` defines

src/sagemaker/tensorflow/README.rst

+1
Original file line numberDiff line numberDiff line change
@@ -482,6 +482,7 @@ Required argument
482482

483483
- ``str``: An S3 URI, for example ``s3://my-bucket/my-training-data``, which indicates the dataset's location.
484484
- ``dict[str, str]``: A dictionary mapping channel names to S3 locations, for example ``{'train': 's3://my-bucket/my-training-data/train', 'test': 's3://my-bucket/my-training-data/test'}``
485+
- ``sagemaker.session.s3_input``: channel configuration for S3 data sources that can provide additional information as well as the path to the training dataset. See `the API docs <https://sagemaker.readthedocs.io/en/latest/session.html#sagemaker.session.s3_input>`_ for full details.
485486

486487
When the training job starts in SageMaker the container will download the dataset.
487488
Both ``train_input_fn`` and ``eval_input_fn`` functions have a parameter called ``training_dir`` which

src/sagemaker/tensorflow/estimator.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -216,8 +216,8 @@ def fit(self, inputs, wait=True, logs=True, job_name=None, run_tensorboard_local
216216
training data, you can specify a dict mapping channel names
217217
to strings or :func:`~sagemaker.session.s3_input` objects.
218218
(sagemaker.session.s3_input) - channel configuration for S3 data sources that can provide
219-
additional information about the training dataset. See :func:`sagemaker.session.s3_input`
220-
for full details.
219+
additional information as well as the path to the training dataset.
220+
See :func:`sagemaker.session.s3_input` for full details.
221221
wait (bool): Whether the call should wait until the job completes (default: True).
222222
logs (bool): Whether to show the logs produced by the job.
223223
Only meaningful when wait is True (default: True).

0 commit comments

Comments
 (0)