understanding-pytorch-batching-lstm
understanding-pytorch-batching-lstm copied to clipboard
Error when running the notebook
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.
Hello Irene, sure I will look into it. I guess that the pytorch version changed a lot so it may be the issue here.