Skip to content

Commit bf826a5

Browse files
authored
doc: remove hardcoded list of target devices in compile() (#1289)
1 parent 6dc9982 commit bf826a5

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/sagemaker/estimator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -484,9 +484,9 @@ def compile_model(
484484
485485
Args:
486486
target_instance_family (str): Identifies the device that you want to
487-
run your model after compilation, for example: ml_c5. Allowed
488-
strings are: ml_c5, ml_m5, ml_c4, ml_m4, jetsontx1, jetsontx2,
489-
ml_p2, ml_p3, deeplens, rasp3b
487+
run your model after compilation, for example: ml_c5. For allowed
488+
strings see
489+
https://docs.aws.amazon.com/sagemaker/latest/dg/API_OutputConfig.html.
490490
input_shape (dict): Specifies the name and shape of the expected
491491
inputs for your trained model in json dictionary form, for
492492
example: {'data':[1,3,1024,1024]}, or {'var1': [1,1,28,28],

src/sagemaker/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -299,9 +299,9 @@ def compile(
299299
300300
Args:
301301
target_instance_family (str): Identifies the device that you want to
302-
run your model after compilation, for example: ml_c5. Allowed
303-
strings are: ml_c5, ml_m5, ml_c4, ml_m4, jetsontx1, jetsontx2,
304-
ml_p2, ml_p3, deeplens, rasp3b
302+
run your model after compilation, for example: ml_c5. For allowed
303+
strings see
304+
https://docs.aws.amazon.com/sagemaker/latest/dg/API_OutputConfig.html.
305305
input_shape (dict): Specifies the name and shape of the expected
306306
inputs for your trained model in json dictionary form, for
307307
example: {'data':[1,3,1024,1024]}, or {'var1': [1,1,28,28],

0 commit comments

Comments
 (0)