Skip to content

Commit 3e463af

Browse files
authored
TF ZCC limitation footnote (aws#342)
1 parent 6f8e757 commit 3e463af

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ The following frameworks are available AWS Deep Learning Containers with the dee
6868
| [PyTorch](docs/pytorch.md) | 1.4, 1.5, 1.6 |
6969
| [XGBoost](docs/xgboost.md) | 0.90-2, 1.0-1 ([As a built-in algorithm](docs/xgboost.md#use-xgboost-as-a-built-in-algorithm))|
7070

71+
**Note**: Debugger with zero script change is partially available for TensorFlow v2.1.0 and v2.3.0. The `inputs`, `outputs`, `gradients`, and `layers` built-in collections are currently not available for these TensorFlow versions.
72+
7173
### AWS training containers with script mode
7274

7375
The `smdebug` library supports frameworks other than the ones listed above while using AWS containers with script mode. If you want to use SageMaker Debugger with one of the following framework versions, you need to make minimal changes to your training script.

docs/tensorflow.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,14 @@ tf_estimator = TensorFlow(
6464
)
6565
tf_estimator.fit("s3://bucket/path/to/training/data")
6666
```
67+
>**Note**: The SageMaker TensorFlow estimator and the Debugger collections in the example are based on the SageMaker python SDK v2 and `smdebug` v0.9.2. It is highly recommended to upgrade the packages by executing the following command line.
68+
```bash
69+
pip install -U sagemaker
70+
pip install -U smdebug
71+
```
72+
If you are using Jupyter Notebook, put exclamation mark at the front of the code lines and restart your kernel.
73+
74+
#### Available Tensor Collections for TensorFlow
6775

6876
**Note**: The SageMaker TensorFlow estimator and the Debugger collections in this example are based on the latest `smdebug` library. We highly recommend that you upgrade the packages by running the following commands at the command line:
6977
```

0 commit comments

Comments
 (0)