-
Notifications
You must be signed in to change notification settings - Fork 27
Add SageMaker Elastic Inference test #21
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
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.
Corresponding readme changes :D
updated with the content from https://github.com/aws/sagemaker-mxnet-container/blob/a542b9c4783b7ec567e341b66de82cd9b774ec12/README.rst |
README.rst
Outdated
pytest test/integration/sagemaker --aws-id 12345678910 \ | ||
--docker-base-name preprod-mxnet \ | ||
--instance-type ml.m4.xlarge \ | ||
--tag 1.3.0-cpu-py3 |
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.
1.4.0
README.rst
Outdated
|
||
# Example | ||
pytest test/integration/sagemaker --aws-id 12345678910 \ | ||
--docker-base-name preprod-mxnet \ |
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.
mxnet-serving
README.rst
Outdated
|
||
# Example for running Elastic Inference SageMaker end to end test | ||
pytest test/integration/sagemaker/test_elastic_inference.py --aws-id 12345678910 \ | ||
--docker-base-name preprod-mxnet \ |
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.
same
README.rst
Outdated
|
||
:: | ||
|
||
pytest test/integration/sagmaker --aws-id <your_aws_id> \ |
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.
tox or pytest?
Description of changes:
recreation of #11. I chose to use a smaller model tar that was already in the repository so that we're not managing large binaries with git and so that we don't run into the error that #11 did with the Python SDK's model repacking functionality.
The test passed when run from my laptop.
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.