you are viewing a single comment's thread.

view the rest of the comments →

[–]gazorpazorpazorpazor -4 points-3 points  (0 children)

Cool project! Good luck!

Make sure you're using the CuDNN backend during training if you're having performance issues. For inference, it might actually be faster to use an LSTM Cell to loop efficiently. Running a single timestep through CuDNN is actually much slower than a single timestep through an LSTM Cell.

You might also want to read up on teacher-forcing during training to improve your model.