Skip to content

Conversation

@nadiaya
Copy link
Contributor

@nadiaya nadiaya commented Jun 20, 2018

Issue #53

Add support for PyTorch framework.

Merge Checklist

Put an x in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.

  • I have read the CONTRIBUTING doc
  • I have added tests that prove my fix is effective or that my feature works (if appropriate)
  • I have updated the changelog with a description of my changes (if appropriate)
  • I have updated any necessary documentation (if appropriate)

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

nadiaya added 30 commits March 27, 2018 22:12
Fix whitespaces.
Make python3 to be default for pytorch estimator.
Add Pytorch estimator and model
…sion based on the current environment python version.
Add integ tests for pytorch prediction.
Add support for pytorch 0.4.0.
Make 0.4 to be default version. Get rid of 03.1 support.
* Switch pytorch estimator integ tests to use script mode. Replace double quotes with single quotes.

* Switch failure script to use script mode.
* Add initial Readme commit for PyTorch.

* Update table of contents. Readme updates.

* Add more info about distributed training.

* PR comments.

* Add notes about argparse actions and how to use boolean hyperparameters.
@nadiaya nadiaya requested review from owen-t and winstonaws June 20, 2018 16:41
@codecov-io
Copy link

codecov-io commented Jun 20, 2018

Codecov Report

Merging #243 into master will increase coverage by 0.11%.
The diff coverage is 98.36%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #243      +/-   ##
==========================================
+ Coverage   92.18%   92.29%   +0.11%     
==========================================
  Files          45       49       +4     
  Lines        3197     3257      +60     
==========================================
+ Hits         2947     3006      +59     
- Misses        250      251       +1
Impacted Files Coverage Δ
src/sagemaker/fw_utils.py 98.52% <100%> (ø) ⬆️
src/sagemaker/pytorch/estimator.py 100% <100%> (ø)
src/sagemaker/pytorch/__init__.py 100% <100%> (ø)
src/sagemaker/pytorch/defaults.py 100% <100%> (ø)
src/sagemaker/pytorch/model.py 96.55% <96.55%> (ø)

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update b8f00ff...276082e. Read the comment docs.

Your PyTorch training script must be a Python 2.7 or 3.5 compatible source file.

The training script is very similar to a training script you might run outside of SageMaker, but you
can access useful properties about the training environment through various environment variables, such as
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Saying "such as" kind of makes me think there is a complete list somewhere of all of these variables, but I can't find it. How does a customer get a list of all the environment variables?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's bellow:

For more on training environment variables, please visit SageMaker Containers <https://github.com/aws/sagemaker-containers>_.

@nadiaya nadiaya merged commit 92eb47d into aws:master Jun 20, 2018
@nadiaya nadiaya deleted the pytorch branch June 28, 2018 00:17
knakad pushed a commit to knakad/sagemaker-python-sdk that referenced this pull request Nov 26, 2019
* Rename ScriptModeProcessor to ScriptProcessor

* Move arguments param down to run method in Processor classes

* Remove py_version from ScriptProcessor and add command as a param to ScriptProcessor.run

* Remove Enums in favor of string literals
knakad pushed a commit to knakad/sagemaker-python-sdk that referenced this pull request Dec 4, 2019
* Rename ScriptModeProcessor to ScriptProcessor

* Move arguments param down to run method in Processor classes

* Remove py_version from ScriptProcessor and add command as a param to ScriptProcessor.run

* Remove Enums in favor of string literals
knakad pushed a commit that referenced this pull request Dec 4, 2019
* Rename ScriptModeProcessor to ScriptProcessor

* Move arguments param down to run method in Processor classes

* Remove py_version from ScriptProcessor and add command as a param to ScriptProcessor.run

* Remove Enums in favor of string literals
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants