a4697d7e45e66a3264eb56dcf489d67d4df40d23,implementations/cgan/cgan.py,Generator,forward,#Generator#,66

Before Change

    def forward(self, noise, labels):
        gen_input = torch.mul(self.label_emb(labels), noise)
        out = self.l1(gen_input)
        out = out.view(out.shape[0], 128, self.init_size, self.init_size)
        img = self.model(out)
        return img

class Discriminator(nn.Module):

After Change

    def forward(self, noise, labels):
        gen_input = torch.mul(self.label_emb(labels), noise)
        img = self.model(gen_input)
        img = img.view(img.size(0), opt.channels, opt.img_size, opt.img_size)
        return img

class Discriminator(nn.Module):
    def __init__(self):

Instances