Commit 8094b336 authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Added dropout for RNN layers

parent a3d79adf
Pipeline #226523 failed with stages
in 32 seconds
......@@ -86,6 +86,7 @@ public class AllPredefinedLayers {
public static final String BEAMSEARCH_MAX_LENGTH = "max_length";
public static final String BEAMSEARCH_WIDTH_NAME = "width";
public static final String SHAPE_NAME = "shape";
public static final String RNN_DROPOUT_NAME = "dropout";
//possible String values
public static final String PADDING_VALID = "valid";
......
......@@ -101,6 +101,11 @@ abstract public class BaseRNN extends PredefinedLayerDeclaration {
.constraints(Constraints.INTEGER, Constraints.POSITIVE)
.defaultValue(1)
.build(),
new ParameterSymbol.Builder()
.name(AllPredefinedLayers.RNN_DROPOUT_NAME)
.constraints(Constraints.NUMBER, Constraints.BETWEEN_ZERO_AND_ONE)
.defaultValue(0)
.build(),
new ParameterSymbol.Builder()
.name(AllPredefinedLayers.BIDIRECTIONAL_NAME)
.constraints(Constraints.BOOLEAN)
......
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