Thank you for translating the time series example to Lightning. I’ve run this code and find that the MSE loss steadily improves (decreases) up to the 4th epoch, getting down to < 1e-4. During the 5th epoch, the loss U-turns and rapidly explodes to > 1e+20 and never recovers for the subsequent epochs.
Wondering if you saw something similar? I’m trying to learn pytorch lightning for the first time so I’m trying to to figure out if it is a problem with the original pytorch example, with the translation to lightning, or with the translation to my code (the last seems unlikely because I tried directly copy-and-pasting your code and still got the same result)
Thanks!