关于数据集规模和特征生成的一些问题

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

关于数据集规模和特征生成的一些问题 #17

Open rainman503 opened 1 year ago

rainman503 commented 1 year ago

作者你好，非常感谢你的工作！！但是我还是有几个问题想请教一下：1、我想拿扩散模型来做特征生成（1×4096d的特征），但是不知道为什么效果并不是很好，所生成的特征和原始特征分布差异非常大。作者有什么好的建议吗？2、训练扩散模型的数据集是否需要非常大规模的数据集？在小规模数据集（比如：1k左右）上模型的生成质量是不是会大打折扣？