91df825459fd2b6ba587e20730a06333482d7b65,onmt/modules/SRU.py,SRUCell,get_dropout_mask_,#SRUCell#Any#Any#,544

Before Change

    def get_dropout_mask_(self, size, p):
        w = self.weight.data
        return Variable(w.new(*size).bernoulli_(1-p).div_(1-p))


class SRU(nn.Module):
    

After Change

    def get_dropout_mask_(self, size, p):
        w = self.weight.data
        return w.new(*size).bernoulli_(1-p).div_(1-p)


class SRU(nn.Module):
    

Instances