Unet error in DDPM

[TwinkleStarst]

When training with the celeba dataset, the program encountered an error.

[Paschalis-D]

Yes, I am getting the same one. Did you find any solution? I'll let you know if I do.

[baizhu0414]

same issue.Have you solved the problem?
When I use dataset MNIST,it works well.

[NisargUpadhyayIITJ]

I'd like to work on this issue. Can I take it?

[G23yl]

DownSample use Conv2d with kernel_size (3, 3), but UpSample use ConvTranspose2d with kernel_size (4, 4) which results in the (H, W) of x larger than s. I add a F.pad operation before torch.cat