Examples on how to use different frameworks on SageMaker.
Apache MXNet¶
- MNIST Training with MXNet and Gluon
- Download training and test data
- Uploading the data
- Implement the training function
- Run the training script on SageMaker
- Cleanup
- Building an image embedding server with Gluon
- Sentiment Analysis with Apache MXNet and Gluon
- Implementing a Recommender System with SageMaker, MXNet, and Gluon
- Importing and hosting an ONNX model with MXNet
- Exporting ONNX Models with MXNet
- Using the Apache MXNet Module API with SageMaker Training and Batch Transformation
- Using Amazon Elastic Inference with MXNet on Amazon SageMaker
Deep Graph Library¶
- Graph convolutional matrix completion hyperparameter tuning with Amazon SageMaker and Deep Graph Library with MXNet backend
- Training Amazon SageMaker models for molecular property prediction by using DGL with PyTorch backend
- Hyperparameter tuning with Amazon SageMaker for molecular property prediction
- Training Amazon SageMaker models by using the Deep Graph Library with MXNet backend
- Output
- Hyperparameter tuning with Amazon SageMaker and Deep Graph Library with MXNet backend
- Training Amazon SageMaker models by using the Deep Graph Library with PyTorch backend
- Output
- Hyperparameter tuning with Amazon SageMaker and Deep Graph Library with PyTorch backend
- Training knowledge graph embedding by using the Deep Graph Library with MXNet backend
- Output
- Hyperparameter tuning with Amazon SageMaker and Deep Graph Library with MXNet backend
- Training knowledge graph embedding by using the Deep Graph Library with PyTorch backend
- Output
- Hyperparameter tuning with Amazon SageMaker and Deep Graph Library with PyTorch backend
Scikit-learn¶
TensorFlow¶
- Migrating scripts from Framework Mode to Script Mode
- Construct an entry point script for training
- Deploy the trained model to prepare for predictions
- Invoking the endpoint
- Clean-up
- Horovod Distributed Training with SageMaker TensorFlow script mode.
- TensorFlow Script Mode with Pipe Mode Input
- Running training using the Python SDK
- Using TensorFlow Scripts in SageMaker - Quickstart
- Training in SageMaker
- TensorFlow Script Mode - Using Shell scripts
- Getting the image for training
- Training in SageMaker
- TensorFlow Eager Execution with Amazon SageMaker Script Mode and Automatic Model Tuning
- Prepare dataset
- TensorFlow script mode training and serving
- Set up the environment
- Construct a script for distributed training
- Create a training job using the
TensorFlowestimator - Deploy the trained model to an endpoint
- Invoke the endpoint
- Delete the endpoint
- Visualize Amazon SageMaker Training Jobs with TensorBoard
- TensorBoard
- Cleaning up