Skip to content

ngarneau/understanding-pytorch-batching-lstm

Repository files navigation

Understanding and visualizing PyTorch Batching with LSTM

This is a small notebook that I wrote to help me understand how batching was done in PyTorch with an Recurrent Neural Network (LSTM).

Please, if you see anything wrong within this notebook feel free to contribute or submit an issue, I may have misunderstood/misinterpreted/misrepresented some things here.

The point here wasn't to build a state of the art model but visualize properly how PyTorch handle the tensors while batching them into an LSTM.

Thanks to Tushar-N from which I inspired this repo and of cours the Classifying Names with a Character-Level RNN PyTorch tutorial.

About

Understanding and visualizing PyTorch Batching with LSTM

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published