Skip to content

Commit 1225582

Browse files
bhanutejagkBhanu Teja Goshikonda
andauthored
TF2.19 SageMaker Training Autopatch (#5012)
* added core_packages.json file and building autopatch * corrected path for dlc-pr-tensorflow-2-training in toml file. * removed ',' at end of corepackages files as part of syntax corrections * rebuilding the image with one test enabled * added debugging statements to find root cause of the build timeout. * added debugging to find timeout issue * added extra debugging statements * added debugging for trigger_language_patching * added core_packages.json files * enabled furthur testing * set do_build to false to focus on furthur testing instead of building image again * formatted the code with black and reverted back toml file * reverted back patch_helper and utils files used for debugging * reformatted the files patch_helper and utils --------- Co-authored-by: Bhanu Teja Goshikonda <bhanugk@amazon.com>
1 parent 09b9ae6 commit 1225582

File tree

3 files changed

+77
-1
lines changed

3 files changed

+77
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
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: "True"
8+
autopatch_build: "True"
99

1010
repository_info:
1111
training_repository: &TRAINING_REPOSITORY
Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
{
2+
"awscli": {
3+
"version_specifier": "<2"
4+
},
5+
"opencv-python": {
6+
"version_specifier": "==4.12.0.88",
7+
"skip": "True"
8+
},
9+
"cython": {
10+
"version_specifier": "<3.0"
11+
},
12+
"pyyaml": {
13+
"version_specifier": ">=6.0,<6.1"
14+
},
15+
"sparkmagic": {
16+
"version_specifier": "<1"
17+
},
18+
"tensorflow-io": {
19+
"version_specifier": "==0.37.*"
20+
},
21+
"tensorflow-text": {
22+
"version_specifier": "==2.19.0",
23+
"skip": "True"
24+
},
25+
"tf-keras": {
26+
"version_specifier": "==2.19.0",
27+
"skip": "True"
28+
},
29+
"tensorflow-hub": {
30+
"version_specifier": "==0.16.1",
31+
"skip": "True"
32+
},
33+
"tf-models-official": {
34+
"version_specifier": "==2.19.1",
35+
"skip": "True"
36+
}
37+
}
Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
{
2+
"awscli": {
3+
"version_specifier": "<2"
4+
},
5+
"opencv-python": {
6+
"version_specifier": "==4.11.0.86"
7+
},
8+
"cython": {
9+
"version_specifier": "<3.0"
10+
},
11+
"pyyaml": {
12+
"version_specifier": ">=6.0,<6.1"
13+
},
14+
"sagemaker-experiments": {
15+
"version_specifier": "==0.*"
16+
},
17+
"sparkmagic": {
18+
"version_specifier": "<1"
19+
},
20+
"tensorflow-hub": {
21+
"version_specifier": ">=0.16.1",
22+
"skip": "True"
23+
},
24+
"tensorflow-io": {
25+
"version_specifier": "==0.37.*"
26+
},
27+
"tf-models-official": {
28+
"version_specifier": "==2.19.1",
29+
"skip": "True"
30+
},
31+
"tensorflow-text": {
32+
"version_specifier": "==2.19.0",
33+
"skip": "True"
34+
},
35+
"tf-keras": {
36+
"version_specifier": "==2.19.0",
37+
"skip": "True"
38+
}
39+
}

0 commit comments

Comments
 (0)