Thank you!! It is now added in the list model.named_parameters().
However my collab notebook crashes when I train the LSTM. I thought that there might be a device issue, but both self.initial_h and self.initial_c are on cuda.