aws / sagemaker-training-toolkit

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.
Apache License 2.0
496 stars 118 forks source link

feature: configure IF_NAME for SMDATAPARALLEL #100

Closed yselivonchyk closed 3 years ago

yselivonchyk commented 3 years ago

feature: configure IF_NAME to be used by SMDATAPARALLEL

Forward SageMaker network_if_name to be used by SMDATAPARALLEL as an env variable over mpi launch

Issue #, if available:

Description of changes: Forward SageMaker network interface configuration to Dataparallel library

Testing done:

General

Tests

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

sagemaker-bot commented 3 years ago

AWS CodeBuild CI Report

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository