mirror of
https://github.com/JoergFranke/ADNC.git
synced 2024-11-17 13:58:03 +08:00
bugfix in config
This commit is contained in:
parent
79816892db
commit
8054cc6dda
@ -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'
|
||||||
|
Loading…
Reference in New Issue
Block a user