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: 7 days 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 (almost 7 years ago)
• Default Branch: master
• Last Pushed: 2023-10-26T14:40:33.000Z (about 1 year ago)
• Last Synced: 2024-12-16T13:17:15.717Z (11 days ago)
• Topics: aws, docker, pytorch, sagemaker
• Language: Python
• Homepage:
• Size: 44.8 MB
• Stars: 200
• Watchers: 61
• Forks: 87
• Open Issues: 12
• Metadata Files:
  • Readme: README.rst
  • Changelog: CHANGELOG.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Code of conduct: CODE_OF_CONDUCT.md

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/