diff --git a/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4 b/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4
index 7b10f682fac6767f3e4f9d013721b6a00f8975eb..7db3d4cdd5f5bcd10e3e6e7baead5e474549d87c 100644
--- a/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4
+++ b/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4
@@ -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;
diff --git a/src/main/java/de/monticore/lang/monticar/cnntrain/_cocos/ParameterAlgorithmMapping.java b/src/main/java/de/monticore/lang/monticar/cnntrain/_cocos/ParameterAlgorithmMapping.java
index 1bf00438c9aa1650c108bd5190c826f358ef40eb..a5120206657676b61101e538d734ddad5aa6ef53 100644
--- a/src/main/java/de/monticore/lang/monticar/cnntrain/_cocos/ParameterAlgorithmMapping.java
+++ b/src/main/java/de/monticore/lang/monticar/cnntrain/_cocos/ParameterAlgorithmMapping.java
@@ -51,7 +51,7 @@ class ParameterAlgorithmMapping {
         ASTBatchAxisEntry.class,
         ASTFromLogitsEntry.class,
         ASTIgnoreIndicesEntry.class,
-        //ASTDiceWeightEntry.class,
+        ASTDiceWeightEntry.class,
         ASTMarginEntry.class,
         ASTLabelFormatEntry.class,
         ASTRhoEntry.class,
diff --git a/src/test/resources/valid_tests/FullConfig3.cnnt b/src/test/resources/valid_tests/FullConfig3.cnnt
index ad33493069ba3b860d924ad1ba8c37983373fc82..9270cd3f6895c559f8992d939d623abd51166af5 100644
--- a/src/test/resources/valid_tests/FullConfig3.cnnt
+++ b/src/test/resources/valid_tests/FullConfig3.cnnt
@@ -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