Open snoop2head opened 2 years ago
why wouldn't it work? what's the full error you are seeing?
It seems like num_layers and fmap size defined in two different classes(VQGanVAE and DALLE) caused the problem.
For the fmap size and num_layers in vae.py, it is stated as the following:
Whereas in dalle_pytorch.py, image_fmap_size is the following:
Code-wise they should coincide with the one another, but error log showed that it doesn't.
Will attach the log and code here for the reference.
https://github.com/lucidrains/DALLE-pytorch/blob/4a7958dc0c3313dee91dda5603741139ee5483e3/dalle_pytorch/vae.py#L178