Dropout for training was always disabled
Merge pull request #2 from rmanor/patch-1

Train and test branches for dropout were swop