bugfix in config

This commit is contained in:
Joerg Franke 2018-07-07 15:46:34 +02:00
parent 79816892db
commit 8054cc6dda

View File

@ -29,7 +29,7 @@ mann:
input_embedding: True input_embedding: True
architecture: 'uni' architecture: 'uni'
controller_config: {"num_units":[512], "layer_norm":True, "activation":'tanh', 'cell_type':'clstm', 'connect':'sparse', 'attention':False} controller_config: {"num_units":[512], "layer_norm":True, "activation":'tanh', 'cell_type':'clstm', 'connect':'sparse', 'attention':False}
memory_unit_config: {"cell_type":'cmu', "memory_length":192, "memory_width":128, "read_heads":4, "write_heads": 1, "layer_norm":True, "bypass_dropout":0.9} memory_unit_config: {"cell_type":'cmu', "memory_length":192, "memory_width":128, "read_heads":4, "write_heads": 1, "dnc_norm":True, "bypass_dropout":0.9}
output_function: "softmax" output_function: "softmax"
output_mask: True output_mask: True
loss_function: 'cross_entropy' loss_function: 'cross_entropy'