SageMaker MXNet Training Toolkit
SageMaker MXNet Training Toolkit is an open-source library for using MXNet to train models on Amazon SageMaker. For inference, see SageMaker MXNet Inference Toolkit. For the Dockerfiles used for building SageMaker MXNet Containers, see AWS Deep Learning Containers. For information on running MXNet jobs on Amazon SageMaker, please refer to the SageMaker Python SDK documentation.
Contributing
Please read CONTRIBUTING.md for details on our code of conduct, and the process for submitting pull requests to us.
Testing
Set up a virtual environment for testing.
One of the multiple ways to setup a virtual environment
# use a package virtualenv # create a virtualenv virtualenv -p python3 <name of env> # activate the virtualenv source <name of env>/bin/activate
Install requirements
pip install --upgrade .[test]
Local Test
To run specific test
tox -- -k test/unit/test_training.py::test_train_for_distributed_scheduler
To run an entire file
tox -- test/unit/test_training.py
To run all tests within a folder [e.g. integration/local/]
Note: To run integration tests locally, one needs to build an image. To trigger image build, use -B flag.
tox -- test/integration/local
You can also run them in parallel:
tox -- -n auto test/integration/local
To run for specific interpreter [Python environment], use the -e flag
tox -e py37 -- test/unit/test_training.py
Remote Test
Make sure to provide AWS account ID, Region, Docker base name & Tag. Docker Registry is composed of (aws_id, region) Image URI is composed of (docker_registry, docker_base_name, tag)
Resulting Image URI is composed as: {aws_id}.dkr.ecr.{region}.amazonaws.com/{docker_base_name}:{tag}
tox -- --aws-id <aws_id> --region <region> --docker-base-name <docker_base_name> --tag <tag> test/integration/sagemaker
For more details, refer conftest.py
License
SageMaker MXNet 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/
1 maintainer
Metadata
1 maintainer
Release files for sagemaker-mxnet-training 4.3.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| sagemaker_mxnet_training-4.3.0.tar.gz | 8.8 kB | Details |
Release files / sagemaker_mxnet_training-4.3.0.tar.gz
| Download URL | sagemaker_mxnet_training-4.3.0.tar.gz |
|---|---|
| Size | 8.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
44118bc2d230ce8b1343934cb9e23a1068043c5b476a09ffd33278fe8dd7914e
|
|
BLAKE2b-256 checksum How to use checksums |
7e4d4c8d7f8e74c72f713a4593aadad71d3142d1d19d0c47d02955a1c8c7d8aa
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/40.6.2 requests-toolbelt/0.9.1 tqdm/4.46.1 CPython/3.6.10
|
