Retraining problem

[Eureka-ai-86]

Hello, I would like to use the DBIM model to train on my own dataset. Can I directly utilize the code downloaded from the provided link for training? Is it necessary to first download and train the DDBM model, obtain a trained checkpoint, and then use this checkpoint within the DBIM code before conducting sampling?

[zhengkw18]

You can use our code for training, which includes DDBM-style (--noise_schedule=vp). For difficult datasets like ImageNet, training from scratch is costly, so we initialize from a pretrained diffusion model, and use --noise_schedule=i2sb_cond.