Building a pipeline for training on multiple GPUs with dstack
This repo contains the code to run a training pipeline on multiple GPUs.
The train.py
contains all the machine learning components.
In order to automate the workflow, I use dstack
.
In order to run the workflow, all you need to do is execute the following command in your terminal:
dstack run train-mnist-multi-gpu