Merge pull request #46 from haoransh/master

fix the bugs in keyword argument pass
asyml · Nov 8, 2018 · 9098cb6 · 9098cb6
2 parents dd972bc + 325158a
commit 9098cb6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/texar/modules/decoders/transformer_decoders.py b/texar/modules/decoders/transformer_decoders.py
@@ -239,6 +239,8 @@ def default_hparams():
             "poswise_feedforward": default_transformer_poswise_net_hparams(),
             'multihead_attention': {
                 'num_units': 512,
+                'dropout_rate': 0.1,
+                'output_dim': 512,
                 'num_heads': 8,
             },
             "dim": 512,

diff --git a/texar/modules/encoders/transformer_encoders.py b/texar/modules/encoders/transformer_encoders.py
@@ -234,6 +234,8 @@ def default_hparams():
             'poswise_feedforward': default_transformer_poswise_net_hparams(),
             'multihead_attention': {
                 'num_units': 512,
+                'dropout_rate':0.1,
+                'output_dim': 512,
                 'num_heads': 8,
             },
             'dim': 512,