Skip to content

Commit 1f9def5

Browse files
doc: specify that input_shape in no longer required for torch 2.0 mod… (#4159)
Co-authored-by: Akash Goel <[email protected]>
1 parent ef3d2b2 commit 1f9def5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/model.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1101,7 +1101,9 @@ def compile(
11011101
input_shape (dict): Specifies the name and shape of the expected
11021102
inputs for your trained model in json dictionary form, for
11031103
example: {'data': [1,3,1024,1024]}, or {'var1': [1,1,28,28],
1104-
'var2': [1,1,28,28]}
1104+
'var2': [1,1,28,28]}. Input shape is optional for those models traced with torch 2.0,
1105+
see pytorch section of this document for detail:
1106+
https://docs.aws.amazon.com/sagemaker/latest/dg/neo-compilation-preparing-model.html
11051107
output_path (str): Specifies where to store the compiled model
11061108
role (str): Execution role
11071109
tags (list[dict]): List of tags for labeling a compilation job. For

0 commit comments

Comments
 (0)