Skip to content

Commit 9c2b2d0

Browse files
authored
framework version updates (aws#360)
1 parent d271fbe commit 9c2b2d0

File tree

4 files changed

+10
-13
lines changed

4 files changed

+10
-13
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,11 +64,11 @@ The following frameworks are available AWS Deep Learning Containers with the dee
6464
| Framework | Version |
6565
| --- | --- |
6666
| [TensorFlow](docs/tensorflow.md) | 1.15, 2.1.0, 2.2.0, 2.3.0 |
67-
| [MXNet](docs/mxnet.md) | 1.6 |
67+
| [MXNet](docs/mxnet.md) | 1.6, 1.7 |
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.
71+
**Note**: Debugger with zero script change is partially available for TensorFlow v2.1.0. The `inputs`, `outputs`, `gradients`, and `layers` built-in collections are currently not available for these TensorFlow versions.
7272

7373
### AWS training containers with script mode
7474

@@ -78,7 +78,7 @@ The `smdebug` library supports frameworks other than the ones listed above while
7878
| --- | --- |
7979
| [TensorFlow](docs/tensorflow.md) | 1.13, 1.14, 1.15, 2.1.0, 2.2.0, 2.3.0 |
8080
| Keras (with TensorFlow backend) | 2.3 |
81-
| [MXNet](docs/mxnet.md) | 1.4, 1.5, 1.6 |
81+
| [MXNet](docs/mxnet.md) | 1.4, 1.5, 1.6, 1.7 |
8282
| [PyTorch](docs/pytorch.md) | 1.2, 1.3, 1.4, 1.5, 1.6 |
8383
| [XGBoost](docs/xgboost.md) | 0.90-2, 1.0-1 (As a framework)|
8484

docs/mxnet.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,8 @@
1010

1111
## Support
1212

13-
- Zero Script Change experience where you need no modifications to your training script is supported in the official [SageMaker Framework Container for MXNet 1.6](https://docs.aws.amazon.com/sagemaker/latest/dg/pre-built-containers-frameworks-deep-learning.html), or the [AWS Deep Learning Container for MXNet 1.6](https://aws.amazon.com/machine-learning/containers/).
14-
- This library itself supports the following versions when you use our API which requires a few minimal changes to your training script: MXNet 1.4, 1.5, 1.6.
13+
- Zero Script Change experience where you need no modifications to your training script is supported in the official [AWS Deep Learning Container for MXNet](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#general-framework-containers).
14+
- This library itself supports the following versions when you use our API which requires a few minimal changes to your training script: MXNet 1.4, 1.5, 1.6, and 1.7.
1515
- Only Gluon models are supported
1616
- When the Gluon model is hybridized, inputs and outputs of intermediate layers can not be saved
1717
- Parameter server based distributed training is not yet supported

docs/pytorch.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99

1010
## Support
1111
### Versions
12-
- Zero Script Change experience where you need no modifications to your training script is supported in the official [SageMaker Framework Container for PyTorch 1.3](https://docs.aws.amazon.com/sagemaker/latest/dg/pre-built-containers-frameworks-deep-learning.html), or the [AWS Deep Learning Container for PyTorch 1.3](https://aws.amazon.com/machine-learning/containers/).
12+
- Zero Script Change experience where you need no modifications to your training script is supported in the official [AWS Deep Learning Container for PyTorch](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#general-framework-containers).
1313

14-
- The library itself supports the following versions when using changes to the training script: PyTorch 1.2, 1.3.
14+
- The library itself supports the following versions when using changes to the training script: PyTorch 1.2, 1.3, 1.4, 1.5, and 1.6.
1515

1616
---
1717

docs/tensorflow.md

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,17 +12,14 @@
1212

1313
## Amazon SageMaker Debugger Support for TensorFlow<a name="support"></a>
1414

15-
Amazon SageMaker Debugger python SDK and its client library `smdebug` now fully support TensorFlow 2.2 with the latest version release.
15+
Amazon SageMaker Debugger python SDK and its client library `smdebug` now fully support TensorFlow 2.3 with the latest version release.
1616

17-
- [Amazon SageMaker Python SDK PyPI](https://pypi.org/project/sagemaker/)
18-
- [The latest smdebug PyPI release](https://pypi.org/project/smdebug/)
19-
20-
Using Debugger, you can access tensors of any kind for TensorFlow models, from the Keras model zoo to your own custom model, and save them using Debugger built-in or custom tensor collections. You can run your training script on [the official AWS Deep Learning Containers](https://docs.aws.amazon.com/sagemaker/latest/dg/debugger-container.html) where Debugger can automatically capture tensors from your training job. It doesn't matter whether your TensorFlow models use Keras API or pure TensorFlow API (in eager mode or non-eager mode), you can directly run them on the AWS Deep Learning Containers.
17+
Using Debugger, you can access tensors of any kind for TensorFlow models, from the Keras model zoo to your own custom model, and save them using Debugger built-in or custom tensor collections. You can run your training script on [the official AWS Deep Learning Containers](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#general-framework-containers) where Debugger can automatically capture tensors from your training job. It doesn't matter whether your TensorFlow models use Keras API or pure TensorFlow API (in eager mode or non-eager mode), you can directly run them on the AWS Deep Learning Containers.
2118

2219
Debugger and its client library `smdebug` support debugging your training job on other AWS training containers and custom containers. In this case, a hook registration process is required to manually add the hook features to your training script. For a full list of AWS TensorFlow containers to use Debugger, see [SageMaker containers to use Debugger with script mode](https://docs.aws.amazon.com/sagemaker/latest/dg/train-debugger.html#debugger-supported-aws-containers). For a complete guide for using custom containers, see [Use Debugger in Custom Training Containers](https://docs.aws.amazon.com/sagemaker/latest/dg/debugger-bring-your-own-container.html).
2320

2421
### New Features supported by Debugger
25-
- The latest TensorFlow version fully covered by Debugger is 2.2.0
22+
- The latest TensorFlow version fully covered by Debugger is 2.3.0
2623
- Debug training jobs with the TensorFlow framework or Keras TensorFlow
2724
- Debug training jobs with the TensorFlow eager or non-eager mode
2825
- New built-in tensor collections: `inputs`, `outputs`, `layers`, `gradients`

0 commit comments

Comments
 (0)