diff --git a/experiments/pre_trained/cnn_rc_task/adnc/config.yml b/experiments/pre_trained/cnn_rc_task/adnc/config.yml index 2985725..f18a2e7 100644 --- a/experiments/pre_trained/cnn_rc_task/adnc/config.yml +++ b/experiments/pre_trained/cnn_rc_task/adnc/config.yml @@ -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'