AccuracyIgnoreLabel.cnnt 780 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
/* (c) https://github.com/MontiCore/monticore */
configuration FullConfig{
    num_epoch : 5
    batch_size : 100
    load_pretrained : true
    eval_metric : accuracy_ignore_label{
        axis : 1
        ignore_label : 255
    }
    loss: dice_loss{
    sparse_label: true
    from_logits: true
    loss_axis : -1
    batch_axis : 0
    }
    context : gpu
    normalize : true
    optimizer : rmsprop{
        learning_rate : 0.001
        learning_rate_minimum : 0.00001
        weight_decay : 0.01
        learning_rate_decay : 0.9
        learning_rate_policy : step
        step_size : 1000
        rescale_grad : 1.1
        clip_gradient : 10
        gamma1 : 0.9
        gamma2 : 0.9
        epsilon : 0.000001
        centered : true
        clip_weights : 10
    }
}