Updates to AWS Deep Learning Containers for TensorFlow (1.15.2 & 2.1.0), PyTorch 1.4.0, and MXNet 1.6.0
The AWS Deep Learning Containers are available today with the latest framework versions of TensorFlow 2.1.0 & 1.15.2, PyTorch 1.4.0, and MXNet 1.6.0 . The release includes the addition of Amazon SageMaker Python SDK in the containers, and updates to the Amazon SageMaker Experiments package. Amazon SageMaker Experiments is a feature in Amazon SageMaker that lets you organize, track, compare, and evaluate machine learning (ML) experiments and model versions. The TensorFlow 2.1.0 python3 training containers now also include SageMaker Debugger, which allow data scientists to save and inspect the model tensors during training jobs.