Add optional dropout support

[lecode-official]

In the original paper, the authors performed experiments with and without dropout. Right now, the models do not support dropout, so it should be able to optionally enable dropout during training.