Function marian::inits::dropout

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.


A NodeInitializer

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