Sayed_Nadim:

spectral_norm


Adding SpectralNorm in Linear Layer of ReLULayer works, but when adding it to Linear Layer of the Generator, I got the error.
    def __init__(self, dim=DIM, output_dim=OUTPUT_DIM):
        super(Generator, self).__init__()

        self.dim = dim
        self.ln1 = SpectralNorm(nn.Linear(128, 4*4*8*self.dim))