Skip to content

Commit 4874c6e

Browse files
author
Shotaro Kohama
committed
Add descriptions about resource_config keys
1 parent 7e00b40 commit 4874c6e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/sagemaker/session.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,9 @@ def train(self, image, input_mode, input_config, role, job_name, output_config,
212212
output_config (dict): The S3 URI where you want to store the training results and optional KMS key ID.
213213
resource_config (dict): Contains values for ResourceConfig:
214214
* instance_count (int): Number of EC2 instances to use for training.
215+
The key in resource_config is 'InstanceCount'.
215216
* instance_type (str): Type of EC2 instance to use for training, for example, 'ml.c4.xlarge'.
217+
The key in resource_config is 'InstanceType'.
216218
hyperparameters (dict): Hyperparameters for model training. The hyperparameters are made accessible as
217219
a dict[str, str] to the training code on SageMaker. For convenience, this accepts other types for
218220
keys and values, but ``str()`` will be called to convert them before training.

0 commit comments

Comments
 (0)