Imported from https://github.com/alexboii/Adversarial-Dropout