search

zoubohao / DenoisingDiffusionProbabilityModel-ddpm-

This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.
MIT License
1.48k stars 156 forks source link

关于uNet网络的问题 #42

Open 0208t opened 4 months ago

0208t commented 4 months ago

您好，我在查看网络结构的使用，print了uNet网络的数据大小， image 但程序会出现错误 image xiangwe想问问这个错误有大佬会改进吗？

chenchen278 commented 4 months ago

print('hs.append(h)', hs.append(h))这里出错啦，hs.append(h)的方法会再一次把h插入到hs中，造成后续的维度不匹配。同时该方法调用没有返回值，print了个寂寞hhh。