ImageNet dataset: resize the image or change the dimension of z

[Liang-ZX]

Hi Samarth, thanks for sharing your code.
However, I met a problem when I run this code with ImageNet dataset. The VAE class in model.py hard code the sample shape of z as 1024*4*4, which is not in line with the image size 224*224 of Imagenet. So, I would like to know whether I should resize the image to 32*32 or I should modify the VAE architecture? Thanks.

[devKhush]

Hi Samarth, thanks for sharing your code. However, I met a problem when I run this code with ImageNet dataset. The VAE class in model.py hard code the sample shape of z as 102444, which is not in line with the image size 224224 of Imagenet. So, I would like to know whether I should resize the image to 3232 or I should modify the VAE architecture? Thanks.

Did you find any way to fix this?