Get free access to a cloud GPU if you don't have one with a hosted notebook like Colaboratory or SageMaker StudioLab.

Great, now you are ready to train! 🥳 
Training loop
To keep track of your training progress, use the tqdm library to add a progress bar over the number of training steps: