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 ...@@ -115,7 +115,7 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number
IgnoreIndicesEntry implements SoftmaxCrossEntropyIgnoreIndicesEntry = name:"ignore_indices" ":" value:IntegerValue; IgnoreIndicesEntry implements SoftmaxCrossEntropyIgnoreIndicesEntry = name:"ignore_indices" ":" value:IntegerValue;
MarginEntry implements HingeEntry, SquaredHingeEntry = name:"margin" ":" value:NumberValue; MarginEntry implements HingeEntry, SquaredHingeEntry = name:"margin" ":" value:NumberValue;
LabelFormatEntry implements LogisticEntry = name:"label_format" ":" value:StringValue; 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 OptimizerValue extends ConfigValue;
interface SGDEntry extends OptimizerParamEntry; interface SGDEntry extends OptimizerParamEntry;
......
...@@ -51,7 +51,7 @@ class ParameterAlgorithmMapping { ...@@ -51,7 +51,7 @@ class ParameterAlgorithmMapping {
ASTBatchAxisEntry.class, ASTBatchAxisEntry.class,
ASTFromLogitsEntry.class, ASTFromLogitsEntry.class,
ASTIgnoreIndicesEntry.class, ASTIgnoreIndicesEntry.class,
//ASTDiceWeightEntry.class, ASTDiceWeightEntry.class,
ASTMarginEntry.class, ASTMarginEntry.class,
ASTLabelFormatEntry.class, ASTLabelFormatEntry.class,
ASTRhoEntry.class, ASTRhoEntry.class,
......
...@@ -9,6 +9,7 @@ configuration FullConfig{ ...@@ -9,6 +9,7 @@ configuration FullConfig{
from_logits: true from_logits: true
loss_axis : -1 loss_axis : -1
batch_axis: 0 batch_axis: 0
dice_weight: (0.2, 0.8)
} }
context : gpu context : gpu
normalize : true 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