fc8951bbd36b14099e41fe171ab57e9f3937fe4f,models/wgan.py,LayerNorm,__init__,#LayerNorm#Any#Any#,7

Before Change

OUTPUT_DIM=64*64*3
class LayerNorm(nn.Module):
    def __init__(self, features, eps=1e-5):
        super().__init__()
        self.gamma = nn.Parameter(torch.ones(features))
        self.beta = nn.Parameter(torch.zeros(features))
        self.eps = eps

After Change

        super(ReLULayer, self).__init__()
        self.n_in = n_in
        self.n_out = n_out
        self.linear = nn.Linear(n_in, n_out)
        self.relu = nn.ReLU()

    def forward(self, input):
        output = self.linear(input)

Instances