Commit 5400ae7c authored by Julian Treiber's avatar Julian Treiber

added dice_weight parameter entry

parent 2326a7fb
Pipeline #236229 passed with stages
in 7 minutes and 22 seconds
......@@ -115,7 +115,7 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number
IgnoreIndicesEntry implements SoftmaxCrossEntropyIgnoreIndicesEntry = name:"ignore_indices" ":" value:IntegerValue;
MarginEntry implements HingeEntry, SquaredHingeEntry = name:"margin" ":" value:NumberValue;
LabelFormatEntry implements LogisticEntry = name:"label_format" ":" value:StringValue;
//DiceWeight implements DiceEntry = name: "dice_weight" ":" value:DoubleVectorValue;
DiceWeightEntry implements DiceEntry = name: "dice_weight" ":" value:DoubleVectorValue;
interface OptimizerValue extends ConfigValue;
interface SGDEntry extends OptimizerParamEntry;
......
......@@ -51,7 +51,7 @@ class ParameterAlgorithmMapping {
ASTBatchAxisEntry.class,
ASTFromLogitsEntry.class,
ASTIgnoreIndicesEntry.class,
//ASTDiceWeightEntry.class,
ASTDiceWeightEntry.class,
ASTMarginEntry.class,
ASTLabelFormatEntry.class,
ASTRhoEntry.class,
......
......@@ -9,6 +9,7 @@ configuration FullConfig{
from_logits: true
loss_axis : -1
batch_axis: 0
dice_weight: (0.2, 0.8)
}
context : gpu
normalize : true
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment