GAN-Tutorial icon indicating copy to clipboard operation
GAN-Tutorial copied to clipboard

not an issue: what is the function of the 10 in get_sample_image(G, n_noise)

Open prismspecs opened this issue 5 years ago • 1 comments

def get_sample_image(G, n_noise):
    """
        save sample 100 images
    """
    z = torch.randn(10, n_noise).to(DEVICE)
    y_hat = G(z).view(10, 3, 28, 28).permute(0, 2, 3, 1) # (100, 28, 28)
    result = (y_hat.detach().cpu().numpy()+1)/2.
    return result

What is the deal with the 10? I understand the 100 is random noise as an input, but what about the 10? Thank you so much for your repo btw, very helpful to me.

prismspecs avatar Jul 28 '20 23:07 prismspecs

got same issue: RuntimeError: shape '[10, 3, 28, 28]' is invalid for input of size 30720 site:stackoverflow.com

whgreate avatar Dec 08 '20 05:12 whgreate