Add training parameter to dropout to make it work (#16133)

* Add training parameter to dropout to make it work

I think that without this parameter set dropout is disabled all the time. At least this is what I read in the documentation, besides adding this improves training.

* Removing redundant if statement around dropout

* Fix linter error: line longer than 80.