Peratham / understanding-pytorch-batching-lstm

Understanding and visualizing PyTorch Batching with LSTM

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


Languages

Language:Jupyter Notebook 59.0%Language:Python 41.0%