HI guys,
I was wondering how to remember the hidden and cell states of the LSTM layer across a number of batches. In other words, the last hidden and cell states obtained after training with batch 1 can be used as the initial hidden and cell states for training with batch 2, and so on.
In Luatorch, I think I can use something like model:remember('both'). Is there a similar way to do this in Pytorch? Thank you very much for your help in advance.
Cheers