Adaptar o PyTorch de nó único à aprendizagem profunda distribuída

Saiba como realizar o treinamento distribuído do modelo de aprendizado de máquina usando PyTorch.

Este Notebook segue o fluxo de trabalho de desenvolvimento recomendado. Primeiro, ele mostra como treinar o modelo em um único nó e, em seguida, como adaptar o código usando o HorovodRunner para treinamento distribuído.

Notebook de exemplo do HorovodRunner PyTorch MNIST

Open notebook in new tab