pytorch-seq2seq icon indicating copy to clipboard operation
pytorch-seq2seq copied to clipboard
verified publisher icon IBM

The dimension of predicted_softmax in DecoderRNN.py

Open tk1363704 opened this issue 6 years ago • 0 comments

https://github.com/IBM/pytorch-seq2seq/blob/f146087a9a271e9b50f46561e090324764b081fb/seq2seq/models/DecoderRNN.py#L105 I think .view(batch_size, output_size, -1) should be .view(batch_size, -1, output_size) or this line just makes no sense.

tk1363704 avatar Oct 12 '19 11:10 tk1363704