yangyangii/gan-tutorial

Simple Implementation of many GAN models with PyTorch.

gan-tutorial's Issues

change size of image

Hi, thank you for making an amazing notebook that I can practice.

I would like to change the size of the image.

For example, in R1GAN and BEGAN, IMAGE_DIM is (64, 64, 3). However, I would like to change IMAGE_DIM as (512,512,3) .

I want my result of GAN also to become (512,512,3). Can you give me the instruction about how am I supposed to change the network?

Thank you

Should we freeze parameters in generator / discriminator when training the other?

Hi, thank you for this great repo, it helps me a lot.

I have a simple question about the training process. Should we freeze parameters in generator / discriminator when training the other?

It seems when running G_loss.backward(), parameters in discriminator will change with it?

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

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.