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

← Back to blog

Download the Jupyter Notebook for Long Short-Term Memory with PyTorch + Lightning!!!

A StatQuest Jupyter Notebook

In the StatQuest on Long Short-Term Memory with PyTorch + Lightning we’ll learn how to code an LSTM unit from scratch and then train it. Then we’ll do the same thing with the PyTorch function nn.LSMT(). Along the way we’ll learn two cool tricks that Lightning gives us that make our lives easier: 1) How to add more training epochs without starting over and 2) How to easily visualize the training results to determine if you need to do more training or are done.To give you a head start, you can download the Jupyter Notebook here! TRIPLE BAM!!!