Imported from https://github.com/MSergeyV/Adaptive-dropout-with-Rademacher-complexity