Skip to content

Commit 647759c

Browse files
fix: change default compile model max run to 15 mins (#1305)
1 parent 7dd8c42 commit 647759c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ def compile_model(
476476
output_path,
477477
framework=None,
478478
framework_version=None,
479-
compile_max_run=5 * 60,
479+
compile_max_run=15 * 60,
480480
tags=None,
481481
**kwargs
482482
):

tests/unit/test_mxnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def _create_compilation_job(input_shape, output_location):
160160
"job_name": COMPILATION_JOB_NAME,
161161
"output_model_config": {"S3OutputLocation": output_location, "TargetDevice": "ml_c4"},
162162
"role": ROLE,
163-
"stop_condition": {"MaxRuntimeInSeconds": 300},
163+
"stop_condition": {"MaxRuntimeInSeconds": 900},
164164
"tags": None,
165165
}
166166

0 commit comments

Comments
 (0)