fix sentence encoder from params
@@ -141,9 +141,11 @@ class IntraSentenceAttentionEncoder(Seq2SeqEncoder):
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)
