seems like the this should work as Dropout layer uses self.p as parameter for the nn.functional.dropout function
https://pytorch.org/docs/stable/_modules/torch/nn/modules/dropout.html#Dropout 58
class Dropout(_DropoutNd):
    def forward(self, input):
        return F.dropout(input, self.p, self.training, self.inplace)