search

aws / sagemaker-training-toolkit

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.
Apache License 2.0
496 stars 118 forks source link

P5 instance support #191

Open haozhx23 opened 1 year ago

haozhx23 commented 1 year ago

https://github.com/aws/sagemaker-training-toolkit/blob/e4a765e699e16c5849bbdfd789edbfc9820fdd77/src/sagemaker_training/__init__.py#L87C11-L87C12 As p5 instance is already available for SageMaker training, we expect that p5 instance support appended, thanks!