Commit a0f9a1e2 authored by Sebastian N.'s avatar Sebastian N.

Added dropout for RNNs

parent 66f536fa
Pipeline #226526 failed with stages
in 17 seconds
......@@ -167,6 +167,10 @@ public class ArchitectureElementData {
return getLayerSymbol().getBooleanValue(AllPredefinedLayers.BIDIRECTIONAL_NAME).get();
}
public boolean getDropout() {
return getLayerSymbol().getBooleanValue(AllPredefinedLayers.RNN_DROPOUT_NAME).get();
}
public boolean getFlatten() {
return getLayerSymbol().getBooleanValue(AllPredefinedLayers.FLATTEN_PARAMETER_NAME).get();
}
......
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