denoising-diffusion-pytorch icon indicating copy to clipboard operation
denoising-diffusion-pytorch copied to clipboard

training on CIFAR-10 performs not well, whether L2 loss or FID

Open chengyiqiu1121 opened this issue 8 months ago • 0 comments

hi developers: I am sorry to bother you! I am training on CIFAR-10, with setting train_num_step=500k, batch=16, unet dim 64. But get FID 21.57 not well. And when i increase the train step to 1000k, the FID increases to 23. when I change the dataset to GTSRB, with train step 500k, it gets FID 7.9, and change dataset to CelebA($64\times 64$), it gets FID 13. That is really confusing. Now i am training the model from scratch again using CIFAR-10 with unet dim 128. The loss (L2) is decreasing very slowly(around 0.05). I would appreciate it if you could help me!

image

chengyiqiu1121 avatar Jun 17 '24 05:06 chengyiqiu1121