pytorch-elastic-example

Пример обучения на GPU с использованием PyTorch Elastic

В этом примере рассмотрена отправка задачи обучения на GPU.

Модель обучается с помощью библиотеки PyTorchна датасете MNIST путем создания и отправки задачи обучения.

Для запуска примера:

1. Создайте новый Jupyter Server или подключитесь к уже существующему Jupyter Server, используя документацию.

2. Создайте новую папку и загрузите следующие файлы через веб-интерфейс Jupyter Server на платформе ML Space:

   • mnist.npz— датасет с рукописными цифрами;
   • train_pt-elastic.ipynb — Jupyter-ноутбук для запуска задачи обучения
   • train_ddp_elastic_example-torch2.py — код модели на Pytorch.

3. Запустите ноутбук train_pt_elastic.ipynb в интерфейсе Jupyter Server.