Function marian::inits::dropout

• Defined in File node_initializers.cpp

Function Documentation

Ptr<NodeInitializer> marian::inits::dropout(float dropoutProbability)

Initialize a dropout mask (a tensor of 0 and 1) with given dropout probability.

Dropout is proposed as a technique to prevent Neural Networks from overfitting.

Return

A NodeInitializer

Parameters

• dropoutProbability: a float type defines the dropout probability. E.g., dropoutProbability=0.1 means 90% of values are kept.