https://github.com/aws/sagemaker-pytorch-training-toolkit
Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.
https://github.com/aws/sagemaker-pytorch-training-toolkit
aws docker pytorch sagemaker
Last synced: 3 months ago
JSON representation
Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.
- Host: GitHub
- URL: https://github.com/aws/sagemaker-pytorch-training-toolkit
- Owner: aws
- License: apache-2.0
- Created: 2018-03-01T17:30:09.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2025-02-11T01:29:01.000Z (over 1 year ago)
- Last Synced: 2025-05-08T00:08:04.843Z (about 1 year ago)
- Topics: aws, docker, pytorch, sagemaker
- Language: Python
- Homepage:
- Size: 44.8 MB
- Stars: 203
- Watchers: 59
- Forks: 91
- Open Issues: 12
-
Metadata Files:
- Readme: README.rst
- Changelog: CHANGELOG.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
- Codeowners: CODEOWNERS
Awesome Lists containing this project
README
==================================
SageMaker PyTorch Training Toolkit
==================================
SageMaker PyTorch Training Toolkit is an open-source library for using PyTorch to train models on Amazon SageMaker.
This toolkit depends and extends the base `SageMaker Training Toolkit `__ with PyTorch specific support.
For inference, see `SageMaker PyTorch Inference Toolkit `__.
For the Dockerfiles used for building SageMaker PyTorch Containers, see `AWS Deep Learning Containers `__.
For information on running PyTorch jobs on Amazon SageMaker, please refer to the `SageMaker Python SDK documentation `__.
For notebook examples: `SageMaker Notebook
Examples `__.
Contributing
------------
Please read
`CONTRIBUTING.md `__
for details on our code of conduct, and the process for submitting pull
requests to us.
License
-------
SageMaker PyTorch Training Toolkit is licensed under the Apache 2.0 License. It is copyright 2018 Amazon
.com, Inc. or its affiliates. All Rights Reserved. The license is available at:
http://aws.amazon.com/apache2.0/