Sagemaker-pytorch-training

Latest version: v2.8.1

Safety actively analyzes 666166 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 2 of 7

2.6.0

Features

* add support for native PyTorch DDP distribution

2.5.1

Bug Fixes and Other Changes

* deriver master node from training environment

2.5.0

Features

* Add Heterogeneous Cluster support

2.4.1

Bug Fixes and Other Changes

* CI

2.4.0

Features

* add data parallelism support (11) (12)

Bug Fixes and Other Changes

* use ubuntu 18.04 base image in dlc gpu image
* remove TODOs in 1.6.0 dlc gpu dockerfile and reduce parameters for data parallel integ test
* use base cuda 11 image for test dlc gpu image
* use 1.6.0 for gpu tests and disable horovod tests
* remove local data parallel integ test
* use sagemaker-training 3.7.0 and enable data parallel integ tests
* patch socket call and update flake8 violations

2.3.0

Features

* Use MPIRunnerType

Bug Fixes and Other Changes

* Update main buildspec to only perform CPU integration tests
* Add GPU and unit test buildspecs
* Pin SageMaker version to less than v2

Documentation Changes

* improve training.py doc style

Page 2 of 7

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.