-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Add support for PyTorch #243
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
… it's not yet supported.
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.
Codecov Report
@@ 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
Continue to review full report at Codecov.
|
| 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 |
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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>_.
* 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
* 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
* 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
Issue #53
Add support for PyTorch framework.
Merge Checklist
Put an
xin 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.By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.