Taming LSTMs: Variable-sized mini-batches and why PyTorch is good for your health

William Falcon
Towards Data Science
6 min readJun 4, 2018

--

If you’ve used PyTorch you have likely experienced euphoria, increased energy and may have even felt like walking in the sun for a bit. Your life feels complete again. That is, until you tried to have variable-sized mini-batches using RNNs.

All hope is not lost. After reading this, you’ll be back to fantasies of you + PyTorch…

--

--

⚡️PyTorch Lightning Creator • PhD Student, AI (NYU, Facebook AI research).