Skip to content

Commit

Permalink
add a note for TF 2.2 limited support (aws#303)
Browse files Browse the repository at this point in the history
Co-authored-by: Miyoung Choi <[email protected]>
Co-authored-by: Nihal Harish <[email protected]>
  • Loading branch information
3 people authored Aug 8, 2020
1 parent 145d43b commit e38f381
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,8 @@ The following frameworks are available AWS Deep Learning Containers with the dee
| [PyTorch](docs/pytorch.md) | 1.4, 1.5 |
| [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))|

>**Note**: Limited support of the zero script change experience for TensorFlow 2.2. The tensor collections `layers`, `inputs`, `outputs`, and `gradients` are currently not available.
### AWS training containers with script mode

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.
Expand Down

0 comments on commit e38f381

Please sign in to comment.