Fix check for dropout config incompatibility w/ some architectures

Fix check for dropout config incompatibility w/ some architectures #459