Fixed argument for Adaptive Softmax Instantiation

Summary: Pull Request resolved: https://github.com/pytorch/fairseq/pull/830

Differential Revision: D15960624

Pulled By: myleott

fbshipit-source-id: ecfef5c51b886e3162bb8e07d232c6e9ea1169b0

fairseq/models/lstm.py

@@ -351,7 +351,7 @@ class LSTMDecoder(FairseqIncrementalDecoder):
             self.additional_fc = Linear(hidden_size, out_embed_dim)
         if adaptive_softmax_cutoff is not None:
             # setting adaptive_softmax dropout to dropout_out for now but can be redefined
-            self.adaptive_softmax = AdaptiveSoftmax(num_embeddings, embed_dim, adaptive_softmax_cutoff,
+            self.adaptive_softmax = AdaptiveSoftmax(num_embeddings, hidden_size, adaptive_softmax_cutoff,
                                                     dropout=dropout_out)
         elif not self.share_input_output_embed:
             self.fc_out = Linear(out_embed_dim, num_embeddings, dropout=dropout_out)