Thesis/ADNC
1
0
Fork 0
mirror of https://github.com/JoergFranke/ADNC.git synced 2024-11-17 13:58:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

bugfix in config

Browse Source
This commit is contained in:
Joerg Franke 2018-07-07 15:46:34 +02:00
parent 79816892db
commit 8054cc6dda
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
experiments/pre_trained/cnn_rc_task/adnc/config.yml
View File

@ -29,7 +29,7 @@ mann:
input_embedding: True
architecture: 'uni'
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_mask: True
loss_function: 'cross_entropy'