added AdamW Optimizer

parent 9f74da3c
Pipeline #318606 passed with stage
in 3 minutes and 37 seconds
......@@ -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* "}")?;
......
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