Skip to content

Commit 495f0ca

Browse files
authored
doc: Add note about file URLs for Estimator methods in Local Mode (#1433)
1 parent edd10aa commit 495f0ca

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sagemaker/estimator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,8 @@ def __init__(
133133
stored to a default bucket. If the bucket with the specific name
134134
does not exist, the estimator creates the bucket during the
135135
:meth:`~sagemaker.estimator.EstimatorBase.fit` method execution.
136+
file:// urls are used for local mode. For example: 'file://model/'
137+
will save to the model folder in the current directory.
136138
output_kms_key (str): Optional. KMS key ID for encrypting the
137139
training output (default: None).
138140
base_job_name (str): Prefix for training job name when the
@@ -446,7 +448,8 @@ def fit(self, inputs=None, wait=True, logs="All", job_name=None, experiment_conf
446448
inputs (str or dict or sagemaker.session.s3_input): Information
447449
about the training data. This can be one of three types:
448450
449-
* (str) the S3 location where training data is saved.
451+
* (str) the S3 location where training data is saved, or a file:// path in
452+
local mode.
450453
* (dict[str, str] or dict[str, sagemaker.session.s3_input]) If using multiple
451454
channels for training data, you can specify a dict mapping channel names to
452455
strings or :func:`~sagemaker.session.s3_input` objects.

0 commit comments

Comments
 (0)