aws / aws/amazon-sagemaker-examples
[Content Improvement] Update instance types in pytorch_smdataparallel_mnist_demo
- Dominant language
- Jupyter Notebook
- Stars
- 11k
- Forks
- 7k
- Avg merge
- 8h 29m
- Merged PRs (30d)
- 8
Description
**Link to the notebook**
[pytorch_smdataparallel_mnist_demo](https://github.com/aws/amazon-sagemaker-examples/blob/main/training/distributed_training/pytorch/data_parallel/mnist/pytorch_smdataparallel_mnist_demo.ipynb)
**What aspects of the notebook can be improved?**
This notebook is not working anymore with `ml.p3dn.24xlarge` instances : `botocore.exceptions.ClientError: An error occurred (ValidationException) when calling the CreateTrainingJob operation: Unsupported instance type ml.p3dn.24xlarge`
**What are your suggestions?**
Don't suggest `ml.p3dn.24xlarge` as a recommended instance
Contributor guide
Research direction
Open training/distributed_training/pytorch/data_parallel/mnist/pytorch_smdataparallel_mnist_demo.ipynb and search for the ml.p3dn.24xlarge recommendation. Review the surrounding SageMaker training configuration, update the notebook so it no longer recommends that unsupported instance type, and run the affected notebook cells to confirm the example remains usable.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- aws, jupyter-notebook, pytorch
- Domain
- cloud, documentation, machine-learning
- Issue type
- Documentation
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 48/100