Skip to content

Commit 88ea003

Browse files
author
Deng
committed
update docstring
1 parent e847e69 commit 88ea003

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/pytorch/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def prepare_container_def(self, instance_type, accelerator_type=None):
136136
For example, 'ml.p2.xlarge'.
137137
accelerator_type (str): The Elastic Inference accelerator type to
138138
deploy to the instance for loading and making inferences to the
139-
model. Currently unsupported with PyTorch.
139+
model. Currently supported with PyTorch 1.3.1 Python 3.
140140
141141
Returns:
142142
dict[str, str]: A container definition object usable with the

0 commit comments

Comments
 (0)