Show_attend_tell.cnnt 392 Bytes
Newer Older
1 2 3 4
configuration Show_attend_tell{
     num_epoch:2
     batch_size:2
     context:cpu
5
     eval_metric:bleu
6 7 8 9
     loss:softmax_cross_entropy_ignore_indices{
        ignore_indices:2
     }
     use_teacher_forcing:true
10 11
     save_attention_image:true
     optimizer:adam{
12 13
         learning_rate:0.005
         learning_rate_decay:0.9
14 15 16 17
         step_size:1000
         weight_decay:0.0001
     }
}