Merge pull request #97 from haoransh/master

fix alpha of TransformerDecoder examples
asyml · Mar 2, 2019 · fe8373e · fe8373e
2 parents b2a7cef + f79ab7c
commit fe8373e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/transformer/config_model.py b/examples/transformer/config_model.py
@@ -5,7 +5,7 @@
 
 random_seed = 1234
 beam_width = 5
-alpha = 0.6
+length_penalty = 0.6
 hidden_dim = 512
 
 emb = {

diff --git a/examples/transformer/transformer_main.py b/examples/transformer/transformer_main.py
@@ -128,7 +128,7 @@ def main():
         memory=encoder_output,
         memory_sequence_length=encoder_input_length,
         beam_width=beam_width,
-        alpha=config_model.alpha,
+        length_penalty=config_model.length_penalty,
         start_tokens=start_tokens,
         end_token=eos_token_id,
         max_decoding_length=config_data.max_decoding_length,