I have used the code from https://github.com/pytorch/examples/blob/master/time_sequence_prediction/train.py 10 and modified for my use, but example also used input.chunk() to train the data, but we squeeze the outputs anyway to get the original shape; and it is just the matter of use, at least here.
Anyways, As I stated earlier, I have confirmed that after torch.load() itself we get different(as you said when/where we see the difference) weights loaded for the same model. So, I don’t think printing the h_t and c_t at the test time would help, isn’t it?