added saveAttentionImage variable

src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4

@@ -139,6 +139,9 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number
     ClipWeightsEntry implements RmsPropEntry = name:"clip_weights" ":" value:NumberValue;
     RhoEntry implements AdaDeltaEntry,RmsPropEntry,HuberEntry = name:"rho" ":" value:NumberValue;
 
+    // Visual attention Extension
+    SaveAttentionImage implements ConfigEntry = name:"save_attention_image" ":" value:BooleanValue;
+
     // Reinforcement Extensions
     interface MultiParamValue extends ConfigValue;
 

src/main/java/de/monticore/lang/monticar/cnntrain/_cocos/ParameterAlgorithmMapping.java

@@ -47,7 +47,8 @@ class ParameterAlgorithmMapping {
         ASTFromLogitsEntry.class,
         ASTMarginEntry.class,
         ASTLabelFormatEntry.class,
-        ASTRhoEntry.class
+        ASTRhoEntry.class,
+        ASTSaveAttentionImage.class
     );
 
     private static final List<Class> GENERAL_REINFORCEMENT_PARAMETERS = Lists.newArrayList(

src/main/java/de/monticore/lang/monticar/cnntrain/_symboltable/CNNTrainSymbolTableCreator.java

@@ -322,6 +322,14 @@ public class CNNTrainSymbolTableCreator extends CNNTrainSymbolTableCreatorTOP {
                 .map(n -> n.getNumber().get().intValue())
                 .collect(Collectors.toList());
     }
+
+    @Override
+    public void endVisit(ASTSaveAttentionImage node) {
+        EntrySymbol entry = new EntrySymbol(node.getName());
+        entry.setValue(getValueSymbolForBoolean(node.getValue()));
+        addToScopeAndLinkWithNode(entry, node);
+        configuration.getEntryMap().put(node.getName(), entry);
+    }
     
     @Override
     public void visit(ASTLearningMethodEntry node) {