Skip to content

Commit 6def511

Browse files
reset size increase
1 parent 35a275f commit 6def511

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorflow/training/buildspec-2-19-sm.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK tensorflow
55
version: &VERSION 2.19.0
66
short_version: &SHORT_VERSION "2.19"
77
arch_type: x86
8-
autopatch_build: "False"
8+
autopatch_build: "True"
99

1010
repository_info:
1111
training_repository: &TRAINING_REPOSITORY
@@ -36,7 +36,7 @@ images:
3636
BuildTensorflowSageMakerCpuPy310TrainingDockerImage:
3737
<<: *TRAINING_REPOSITORY
3838
build: &TENSORFLOW_CPU_TRAINING_PY3 false
39-
image_size_baseline: &IMAGE_SIZE_BASELINE 14000
39+
image_size_baseline: &IMAGE_SIZE_BASELINE 7500
4040
device_type: &DEVICE_TYPE cpu
4141
python_version: &DOCKER_PYTHON_VERSION py3
4242
tag_python_version: &TAG_PYTHON_VERSION py312
@@ -52,7 +52,7 @@ images:
5252
BuildTensorflowSageMakerGpuPy310Cu125TrainingDockerImage:
5353
<<: *TRAINING_REPOSITORY
5454
build: &TENSORFLOW_GPU_TRAINING_PY3 false
55-
image_size_baseline: &IMAGE_SIZE_BASELINE 21000
55+
image_size_baseline: &IMAGE_SIZE_BASELINE 11998
5656
device_type: &DEVICE_TYPE gpu
5757
python_version: &DOCKER_PYTHON_VERSION py3
5858
tag_python_version: &TAG_PYTHON_VERSION py312

0 commit comments

Comments
 (0)