Skip to content

Commit 47ab5e5

Browse files
committed
feature: update TF framework version support
1 parent 24300c2 commit 47ab5e5

File tree

3 files changed

+571
-14
lines changed

3 files changed

+571
-14
lines changed

src/sagemaker/fw_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
"local_gpu",
6060
)
6161
SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
62-
"tensorflow": ["2.3.0", "2.3.1", "2.4.1"],
62+
"tensorflow": ["2.3.1", "2.3.2", "2.4.1"],
6363
"pytorch": ["1.6.0", "1.7.1", "1.8.0"],
6464
}
6565
SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]

0 commit comments

Comments
 (0)