Skip to content

Commit 2e82eeb

Browse files
author
Ashish Gupta
committed
fix default time for compilation jobs
1 parent db41eba commit 2e82eeb

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sagemaker/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,7 @@ def compile_model(
718718
'onnx', 'xgboost'
719719
framework_version (str): The version of the framework
720720
compile_max_run (int): Timeout in seconds for compilation (default:
721-
3 * 60). After this amount of time Amazon SageMaker Neo
721+
15 * 60). After this amount of time Amazon SageMaker Neo
722722
terminates the compilation job regardless of its current status.
723723
tags (list[dict]): List of tags for labeling a compilation job. For
724724
more, see

src/sagemaker/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -559,7 +559,7 @@ def compile(
559559
role,
560560
tags=None,
561561
job_name=None,
562-
compile_max_run=5 * 60,
562+
compile_max_run=15 * 60,
563563
framework=None,
564564
framework_version=None,
565565
target_platform_os=None,
@@ -588,7 +588,7 @@ def compile(
588588
https://docs.aws.amazon.com/sagemaker/latest/dg/API_Tag.html.
589589
job_name (str): The name of the compilation job
590590
compile_max_run (int): Timeout in seconds for compilation (default:
591-
3 * 60). After this amount of time Amazon SageMaker Neo
591+
15 * 60). After this amount of time Amazon SageMaker Neo
592592
terminates the compilation job regardless of its current status.
593593
framework (str): The framework that is used to train the original
594594
model. Allowed values: 'mxnet', 'tensorflow', 'keras', 'pytorch',

0 commit comments

Comments
 (0)