atharvap

The self.img_size[0]/16 needs to be typecasted to int: `class Generator(nn.Module): def __init__(self, img_size, latent_dim, dim): super(Generator, self).__init__() self.dim = dim self.latent_dim = latent_dim self.img_size = img_size self.feature_sizes = (int(self.img_size[0] /...

Image shape mismatch

2

As you haven't changed dimensions of the image the input image list needs to be (3XHXW) instead of (HXWX3). It works fine when this change is made.