pypi accelerate 0.2.0
v0.2.0 SageMaker launcher

latest releases: 0.30.0, 0.30.0rc0, 0.29.3...
3 years ago

v0.2.0 SageMaker launcher

SageMaker launcher

It's now possible to launch your training script on AWS instances using SageMaker via accelerate launch.

Kwargs handlers

To customize how the different objects used for mixed precision or distributed training are instantiated, a new API called KwargsHandler is added. This allows the user to pass along the kwargs that will be passed to those objects if used (and it is ignored if those are not used in the current setup, so the script can still run on any kind of setup).

Pad across processes

Trying to gather tensors that are not of the same size across processes resulted in a process hang, a new method Accelerator.pad_across_processes has been added to help with that.

  • Add utility to pad tensor across processes to max length #19 (@sgugger )

Various bug fixes

Don't miss a new accelerate release

NewReleases is sending notifications on new releases.