pypi accelerate 0.3.0
v0.3.0 Notebook launcher and multi-node training

latest releases: 1.0.0rc1, 1.0.0rc0, 0.34.2...
3 years ago

v0.3.0 Notebook launcher and multi-node training

Notebook launcher

After doing all the data preprocessing in your notebook, you can launch your training loop using the new notebook_launcher functionality. This is especially useful for Colab or Kaggle with TPUs! Here is an example on Colab (don't forget to select a TPU runtime).

This launcher also works if you have multiple GPUs on your machine. You just have to pass along num_processes=your_number_of_gpus in the call to notebook_launcher.

Multi-node training

Our multi-node training test setup was flawed and the previous releases of 🤗 Accelerate were not working for multi-node distributed training. This is all fixed now and we have ensured to have more robust tests!

Various bug fixes

Don't miss a new accelerate release

NewReleases is sending notifications on new releases.