295f4afbeb2cb52e17af4a7dd41bfbbea8ec4ee5,hypergan/modules/ez_norm.py,EzNorm,__init__,#EzNorm#Any#Any#Any#Any#Any#Any#,6

Before Change

            self.beta = EqualLinear(style_size, content_size, lr_mul=0.01)
        else:
            self.beta = nn.Linear(style_size, content_size)
        self.conv = nn.Conv2d(channels, 1, 1, 1, padding = (0, 0))
        self.dim = dim

    def forward(self, content, style, epsilon=1e-5):

After Change

        if dims == 2:
            self.conv = nn.Conv1d(channels, 1, 1, 1, padding = 0)
        else:
            self.conv = nn.Conv2d(channels, 1, 1, 1, padding = 0)
        self.dim = dim

    def forward(self, content, style, epsilon=1e-5):
        N = content.shape[0]

Instances