A question about the size of the rnn input emb in function forward

[shaunchen284]

I think there might be a wrong in the model.py.

    def forward(self, input, hidden, return_hiddens=False, noise=False):
        emb = self.drop(self.encoder(input.contiguous().view(-1,self.enc_input_size))) # [(seq_len x batch_size) * feature_size]    
        emb = emb.view(-1, input.size(1), self.rnn_hid_size) # [ seq_len * batch_size * feature_size]

where emb is the input of self.rnn, so I think self.rnn_hid_size should be changed to self.rnn_input_size
Looking forward to your reply

[shaunchen284]

original:
emb = emb.view(-1, input.size(1), self.rnn_hid_size)
to:
emb = emb.view(-1, input.size(1), self.rnn_input_size)