From fbceb207e61842ac60edac40fc151f6b4414f32c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Julian=20Steinsberger-D=C3=BChr=C3=9Fen?= Date: Wed, 12 Aug 2020 05:56:49 +0200 Subject: [PATCH] added AdamW Optimizer --- src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4 | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4 b/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4 index 62d1791..610834c 100644 --- a/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4 +++ b/src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4 @@ -73,6 +73,7 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number | inv:"inv" | poly:"poly" | sigmoid:"sigmoid"); + TrainContextValue implements ConfigValue = (cpu:"cpu" | gpu:"gpu"); interface OptimizerParamEntry extends Entry; @@ -135,6 +136,8 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number interface AdamEntry extends OptimizerParamEntry; AdamOptimizer implements OptimizerValue = name:"adam" ("{" params:AdamEntry* "}")?; + AdamWOptimizer implements OptimizerValue = name:"adamw" ("{" params:AdamEntry* "}")?; + interface RmsPropEntry extends OptimizerParamEntry; RmsPropOptimizer implements OptimizerValue = name:"rmsprop" ("{" params:RmsPropEntry* "}")?; -- GitLab