Skip to content

Commit 028560e

Browse files
author
Talia Chopra
committed
Revert "documentation: removing mention of TF 2.4 from SM distributed model parallel docs"
This reverts commit 8377b86.
1 parent 3f498f4 commit 028560e

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

doc/api/training/smd_model_parallel_release_notes/smd_model_parallel_change_log.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,10 @@
1717

1818
- Adds support for `_register_comm_hook` (PyTorch 1.7 only) which will register the callable as a communication hook for DDP. NOTE: Like in DDP, this is an experimental API and subject to change.
1919

20+
### Tensorflow
21+
22+
- Adds support for Tensorflow 2.4
23+
2024
## Bug Fixes
2125

2226
### PyTorch

doc/api/training/smp_versions/v1.2.0/smd_model_parallel_tensorflow.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
TensorFlow API
22
==============
33

4-
**Supported version: 2.3**
4+
**Supported version: 2.4, 2.3**
55

66
**Important**: This API document assumes you use the following import statement in your training scripts.
77

0 commit comments

Comments
 (0)