Skip to content
This repository was archived by the owner on Dec 16, 2022. It is now read-only.

Commit a4b885c

Browse files
authored
Add scalar_mix_parameters to Elmo.from_params (#1992)
1 parent 1f782d3 commit a4b885c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

allennlp/modules/elmo.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,7 @@ def from_params(cls, params: Params) -> 'Elmo':
212212
do_layer_norm = params.pop_bool('do_layer_norm', False)
213213
keep_sentence_boundaries = params.pop_bool('keep_sentence_boundaries', False)
214214
dropout = params.pop_float('dropout', 0.5)
215+
scalar_mix_parameters = params.pop('scalar_mix_parameters', None)
215216
params.assert_empty(cls.__name__)
216217

217218
return cls(options_file=options_file,
@@ -220,7 +221,8 @@ def from_params(cls, params: Params) -> 'Elmo':
220221
requires_grad=requires_grad,
221222
do_layer_norm=do_layer_norm,
222223
keep_sentence_boundaries=keep_sentence_boundaries,
223-
dropout=dropout)
224+
dropout=dropout,
225+
scalar_mix_parameters=scalar_mix_parameters)
224226

225227

226228
def batch_to_ids(batch: List[List[str]]) -> torch.Tensor:

0 commit comments

Comments
 (0)