Implementing Curriculum Dropout in Pytorch

I am a newbie to Pytorch and I greatly enjoy this community. Is there anyone who can help me implement Curriculum Dropout by Pytorch. Thanks in advance, and any kind of help will be appreciable.
I want to do some experiments of Curriculum Dropout in Pytorch. Curriculum Dropout tries to use a time scheduling for adjusting the dropout rate in the neural networks. The related paper can be downloaded from https://arxiv.org/abs/1703.06229
The source code in python can be found here https://github.com/pmorerio/curriculum-dropout