Lightning AI Studios: Never set up a local environment again →

← Back to blog

Introduction to Coding Neural Networks with PyTorch + Lightning

A StatQuest Jupyter Notebook

Although we saw how to code a simple neural network with PyTorch in The StatQuest Introduction to PyTorch, we can make our lives a lot easier if we add Lightning to the mix. It makes writing the code easier, makes it portable to different computing environments and can even find the learning rate for us! Here we have the the Jupyter Notebook based on the Introduction to Coding Neural Networks with PyTorch + Lightning. TRIPLE BAM!!!