Hi Omer, I am working on something very similar currently (using a modified LSTM network to solve some PDE’s, one of which is Burger’s equation) and I too am struggling to get my pytorch network to converge inline with an identical tf network. If you are still able to recall the problem above it would be great if you could share how you solved the problem in the end, specifically what were the ‘bugs’ in the code and how you were able to fix them.
Many thanks in advance (apologies for any poor grammar or formatting, this is my first post here).