Stage 03 · 03 · Deep learning · skill 2 of 2

PyTorch basics

Tensors, autograd, a training loop you could rewrite from memory.

Recommended first: Neural networks, python-functions

After this you will be able to

• Create tensors and run a forward pass
• Write a training loop with loss.backward() and optimizer.step()
• Move a model to GPU when available

PyTorch · why we picked it