用于创建运行在Amazon SageMaker上的MXNet容器的开源库。
sagemaker-mxnet-training的Python项目详细描述
SageMaker MXNet培训工具包是一个开源库,用于使用MXNet在Amazon SageMaker上培训模型。 有关推断,请参见SageMaker MXNet Inference Toolkit。 有关用于构建SageMaker MXNet容器的Dockerfiles,请参阅AWS Deep Learning Containers。 有关在Amazon SageMaker上运行MXNet作业的信息,请参阅SageMaker Python SDK documentation。在
贡献
请阅读CONTRIBUTING.md 有关我们行为准则的详细信息,以及向我们提交拉取请求的过程。在
测试
设置虚拟环境进行测试。
设置虚拟环境的多种方法之一
# use a package virtualenv # create a virtualenv virtualenv -p python3 <name of env> # activate the virtualenv source <name of env>/bin/activate
安装要求
^{pr2}$局部试验
运行特定测试
tox -- -k test/unit/test_training.py::test_train_for_distributed_scheduler
运行整个文件
tox -- test/unit/test_training.py
运行文件夹中的所有测试[例如integration/local/]
注意:要在本地运行集成测试,需要构建一个映像。要触发图像生成,请使用-B标志。在
tox -- test/integration/local
也可以并行运行:
tox -- -n auto test/integration/local
要为特定的解释器[Python环境]运行,请使用-e标志
tox -e py37 -- test/unit/test_training.py
远程测试
确保提供AWS帐户ID、区域、Docker base名称和标签。 Docker注册表由(aws_id,region)组成 映像URI由(docker_registry,docker_base_name,tag)组成
生成的图像URI组成如下:{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
有关详细信息,请参阅conftest.py
许可证
SageMaker MXNet培训工具包是在Apache2.0许可下授权的。 2018年版权所有亚马逊网站,Inc.或其附属公司。版权所有。 许可证在:http://aws.amazon.com/apache2.0/
- 项目
标签: