fix sentence encoder from params (#963)

allenai · Mar 8, 2018 · 18eb421 · 18eb421
1 parent 9121eed
commit 18eb421
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/allennlp/modules/seq2seq_encoders/intra_sentence_attention.py b/allennlp/modules/seq2seq_encoders/intra_sentence_attention.py
@@ -141,9 +141,11 @@ def from_params(cls, params: Params) -> 'IntraSentenceAttentionEncoder':
         similarity_function = SimilarityFunction.from_params(params.pop('similarity_function', {}))
         num_attention_heads = params.pop_int('num_attention_heads', 1)
         combination = params.pop('combination', '1,2')
+        output_dim = params.pop_int('output_dim', None)
         params.assert_empty(cls.__name__)
         return cls(input_dim=input_dim,
                    projection_dim=projection_dim,
                    similarity_function=similarity_function,
                    num_attention_heads=num_attention_heads,
-                   combination=combination)
+                   combination=combination,
+                   output_dim=output_dim)