DDPM training problem

[AzureCHEM]

Hi,
I'm trying to use celebA-HQ dataset(only images) to train DDPM, and I use unconditional_imgs_flip_256x256.py as the dataset config file,but I always have the following error

The training config file I use is as follows. Thank you so much!

[zeakey]

Reason

Your dataloader provides a dict with key real_img but in your training config you are accessing the img keyword.

https://github.com/open-mmlab/mmgeneration/blob/master/configs/_base_/datasets/unconditional_imgs_flip_256x256.py#L4

https://github.com/open-mmlab/mmgeneration/blob/master/configs/_base_/models/improved_ddpm/ddpm_32x32.py#L43

Solution:

change train_cfg = dict(use_ema=True, real_img_key='img') to train_cfg = dict(use_ema=True, real_img_key='real_img').