Hey @ptrblck,
Thanks for your interest. I don’t purposely do such thing as I use the iterator mechanism on a pre-defined, standard dataset, namely, IMDB. The error might be due to the confusion between the hidden and self.hidden_dim params located in the forward function. When you have some free time, could you please check them?