Hi @tom if I remember correctly, I worked by adding tiny bits of code to a basic working (i.e. non memory exploding) program. I’d tried starting with something complicated and adding a bi-directional LSTM but that just blew up. So did it the other war around and started with a very simple LSTM skeleton.
I hope this helps?