Error when running the notebook

[eirenevp]

Hi,
when I am trying to run the notebook with python 3.6 I keep getting this error message:
RuntimeError: Expected hidden[0] size (1, 32, 32), got (2, 32, 32)
I think the problem starts here, but I am not sure:
---> 26 packed_output, (ht, ct) = self.lstm(packed_input, self.hidden) # Figure 6

It would be very helpful for me if you could fix the error and thus having a running example again.

[ngarneau]

Hello Irene, sure I will look into it. I guess that the pytorch version changed a lot so it may be the issue here.