├── .coveragerc_py310 ├── .coveragerc_py38 ├── .coveragerc_py39 ├── .flake8 ├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── config.yml │ ├── documentation-request.md │ └── feature_request.md └── PULL_REQUEST_TEMPLATE.md ├── .gitignore ├── .pylintrc ├── CHANGELOG.md ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── ENVIRONMENT_VARIABLES.md ├── LICENSE ├── MANIFEST.in ├── NOTICE ├── README.md ├── VERSION ├── branding └── icon │ └── sagemaker-banner.png ├── setup.cfg ├── setup.py ├── src ├── __init__.py └── sagemaker_training │ ├── __init__.py │ ├── _entry_point_type.py │ ├── c │ ├── gethostname.c │ ├── jsmn.c │ └── jsmn.h │ ├── cli │ ├── __init__.py │ └── train.py │ ├── content_types.py │ ├── encoders.py │ ├── entry_point.py │ ├── environment.py │ ├── errors.py │ ├── files.py │ ├── functions.py │ ├── intermediate_output.py │ ├── logging_config.py │ ├── mapping.py │ ├── modules.py │ ├── mpi.py │ ├── params.py │ ├── process.py │ ├── pytorch_xla.py │ ├── record_pb2.py │ ├── recordio.py │ ├── runner.py │ ├── smdataparallel.py │ ├── timeout.py │ ├── torch_distributed.py │ └── trainer.py ├── test ├── __init__.py ├── conftest.py ├── container │ ├── dummy │ │ ├── Dockerfile │ │ ├── requirements.txt │ │ └── train.py │ └── tensorflow │ │ ├── Dockerfile │ │ └── train.py ├── fake_ml_framework.py ├── functional │ ├── __init__.py │ ├── simple_framework.py │ ├── test_download_and_import.py │ ├── test_intermediate_output.py │ ├── test_mpi.py │ └── test_training_framework.py ├── integration │ └── local │ │ ├── test_dummy.py │ │ └── test_tensorflow.py ├── resources │ └── openmpi │ │ ├── Dockerfile │ │ ├── Dockerfile.base │ │ ├── launcher.sh │ │ └── script.py └── unit │ ├── __init__.py │ ├── c │ └── test_gethostname.py │ ├── cli │ ├── __init__.py │ └── test_train.py │ ├── dummy │ ├── __init__.py │ ├── dummy.py │ └── tensorflow │ │ ├── __init__.py │ │ └── compiler │ │ ├── __init__.py │ │ └── xla │ │ ├── __init__.py │ │ └── dummy_xla.py │ ├── test_encoder.py │ ├── test_entry_point.py │ ├── test_entry_point_type.py │ ├── test_environment.py │ ├── test_errors.py │ ├── test_files.py │ ├── test_functions.py │ ├── test_intermediate_output.py │ ├── test_mapping.py │ ├── test_modules.py │ ├── test_mpi.py │ ├── test_process.py │ ├── test_pytorch_xla.py │ ├── test_runner.py │ ├── test_smdataparallel.py │ ├── test_timeout.py │ ├── test_torch_distributed.py │ └── test_trainer.py └── tox.ini /.coveragerc_py310: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.coveragerc_py310 -------------------------------------------------------------------------------- /.coveragerc_py38: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.coveragerc_py38 -------------------------------------------------------------------------------- /.coveragerc_py39: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.coveragerc_py39 -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.github/ISSUE_TEMPLATE/documentation-request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/.pylintrc -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- 1 | * @aws/sagemaker-jobs-platform 2 | -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /ENVIRONMENT_VARIABLES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/ENVIRONMENT_VARIABLES.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 5.1.2.dev0 2 | -------------------------------------------------------------------------------- /branding/icon/sagemaker-banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/branding/icon/sagemaker-banner.png -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [bdist_wheel] 2 | universal=1 3 | 4 | [metadata] 5 | description-file = README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_training/_entry_point_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/_entry_point_type.py -------------------------------------------------------------------------------- /src/sagemaker_training/c/gethostname.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/c/gethostname.c -------------------------------------------------------------------------------- /src/sagemaker_training/c/jsmn.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/c/jsmn.c -------------------------------------------------------------------------------- /src/sagemaker_training/c/jsmn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/c/jsmn.h -------------------------------------------------------------------------------- /src/sagemaker_training/cli/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/sagemaker_training/cli/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/cli/train.py -------------------------------------------------------------------------------- /src/sagemaker_training/content_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/content_types.py -------------------------------------------------------------------------------- /src/sagemaker_training/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/encoders.py -------------------------------------------------------------------------------- /src/sagemaker_training/entry_point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/entry_point.py -------------------------------------------------------------------------------- /src/sagemaker_training/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/environment.py -------------------------------------------------------------------------------- /src/sagemaker_training/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/errors.py -------------------------------------------------------------------------------- /src/sagemaker_training/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/files.py -------------------------------------------------------------------------------- /src/sagemaker_training/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/functions.py -------------------------------------------------------------------------------- /src/sagemaker_training/intermediate_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/intermediate_output.py -------------------------------------------------------------------------------- /src/sagemaker_training/logging_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/logging_config.py -------------------------------------------------------------------------------- /src/sagemaker_training/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/mapping.py -------------------------------------------------------------------------------- /src/sagemaker_training/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/modules.py -------------------------------------------------------------------------------- /src/sagemaker_training/mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/mpi.py -------------------------------------------------------------------------------- /src/sagemaker_training/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/params.py -------------------------------------------------------------------------------- /src/sagemaker_training/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/process.py -------------------------------------------------------------------------------- /src/sagemaker_training/pytorch_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/pytorch_xla.py -------------------------------------------------------------------------------- /src/sagemaker_training/record_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/record_pb2.py -------------------------------------------------------------------------------- /src/sagemaker_training/recordio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/recordio.py -------------------------------------------------------------------------------- /src/sagemaker_training/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/runner.py -------------------------------------------------------------------------------- /src/sagemaker_training/smdataparallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/smdataparallel.py -------------------------------------------------------------------------------- /src/sagemaker_training/timeout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/timeout.py -------------------------------------------------------------------------------- /src/sagemaker_training/torch_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/torch_distributed.py -------------------------------------------------------------------------------- /src/sagemaker_training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/src/sagemaker_training/trainer.py -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/__init__.py -------------------------------------------------------------------------------- /test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/conftest.py -------------------------------------------------------------------------------- /test/container/dummy/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/container/dummy/Dockerfile -------------------------------------------------------------------------------- /test/container/dummy/requirements.txt: -------------------------------------------------------------------------------- 1 | pyfiglet==0.8.post1 2 | -------------------------------------------------------------------------------- /test/container/dummy/train.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/container/tensorflow/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/container/tensorflow/Dockerfile -------------------------------------------------------------------------------- /test/container/tensorflow/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/container/tensorflow/train.py -------------------------------------------------------------------------------- /test/fake_ml_framework.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/fake_ml_framework.py -------------------------------------------------------------------------------- /test/functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/functional/__init__.py -------------------------------------------------------------------------------- /test/functional/simple_framework.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/functional/simple_framework.py -------------------------------------------------------------------------------- /test/functional/test_download_and_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/functional/test_download_and_import.py -------------------------------------------------------------------------------- /test/functional/test_intermediate_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/functional/test_intermediate_output.py -------------------------------------------------------------------------------- /test/functional/test_mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/functional/test_mpi.py -------------------------------------------------------------------------------- /test/functional/test_training_framework.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/functional/test_training_framework.py -------------------------------------------------------------------------------- /test/integration/local/test_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/integration/local/test_dummy.py -------------------------------------------------------------------------------- /test/integration/local/test_tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/integration/local/test_tensorflow.py -------------------------------------------------------------------------------- /test/resources/openmpi/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/resources/openmpi/Dockerfile -------------------------------------------------------------------------------- /test/resources/openmpi/Dockerfile.base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/resources/openmpi/Dockerfile.base -------------------------------------------------------------------------------- /test/resources/openmpi/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/resources/openmpi/launcher.sh -------------------------------------------------------------------------------- /test/resources/openmpi/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/resources/openmpi/script.py -------------------------------------------------------------------------------- /test/unit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/__init__.py -------------------------------------------------------------------------------- /test/unit/c/test_gethostname.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/c/test_gethostname.py -------------------------------------------------------------------------------- /test/unit/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/cli/__init__.py -------------------------------------------------------------------------------- /test/unit/cli/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/cli/test_train.py -------------------------------------------------------------------------------- /test/unit/dummy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/dummy/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/dummy/dummy.py -------------------------------------------------------------------------------- /test/unit/dummy/tensorflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/dummy/tensorflow/compiler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/dummy/tensorflow/compiler/xla/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/dummy/tensorflow/compiler/xla/dummy_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/dummy/tensorflow/compiler/xla/dummy_xla.py -------------------------------------------------------------------------------- /test/unit/test_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_encoder.py -------------------------------------------------------------------------------- /test/unit/test_entry_point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_entry_point.py -------------------------------------------------------------------------------- /test/unit/test_entry_point_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_entry_point_type.py -------------------------------------------------------------------------------- /test/unit/test_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_environment.py -------------------------------------------------------------------------------- /test/unit/test_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_errors.py -------------------------------------------------------------------------------- /test/unit/test_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_files.py -------------------------------------------------------------------------------- /test/unit/test_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_functions.py -------------------------------------------------------------------------------- /test/unit/test_intermediate_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_intermediate_output.py -------------------------------------------------------------------------------- /test/unit/test_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_mapping.py -------------------------------------------------------------------------------- /test/unit/test_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_modules.py -------------------------------------------------------------------------------- /test/unit/test_mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_mpi.py -------------------------------------------------------------------------------- /test/unit/test_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_process.py -------------------------------------------------------------------------------- /test/unit/test_pytorch_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_pytorch_xla.py -------------------------------------------------------------------------------- /test/unit/test_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_runner.py -------------------------------------------------------------------------------- /test/unit/test_smdataparallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_smdataparallel.py -------------------------------------------------------------------------------- /test/unit/test_timeout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_timeout.py -------------------------------------------------------------------------------- /test/unit/test_torch_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_torch_distributed.py -------------------------------------------------------------------------------- /test/unit/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/test/unit/test_trainer.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-training-toolkit/HEAD/tox.ini --------------------------------------------------------------------------------