Skip to content

Lab 5: Troubleshooting & Testing

By Charles Frye. Published August 24, 2022.

In this lab, we'll check out the basic tools required to write clean Python code and see how to write memorization tests for training code in PyTorch Lightning. Then we'll take a deep dive into the trace of a PyTorch training step and use it to debug performance issues in GPU-accelerated code.


  • 00:00 Overview
  • 00:51 Linting: pre-commit, black, flake8
  • 05:42 Testing: pytest, doctest, memorization testing
  • 11:15 Troubleshooting PyTorch performance
  • 16:13 A guided tour of a PyTorch trace

We are excited to share this course with you for free.

We have more upcoming great content. Subscribe to stay up to date as we release it.

We take your privacy and attention very seriously and will never spam you. I am already a subscriber