Training procedure

[CharlesGong12]

Hi, thanks for your amazing work!
I find your work so interesting and I wanna train the models by myself. Could you provide training code?

[vidit98]

Hi, thanks for your interest! It would be difficult to release the training procedure soon. But I can help if you have any specific doubts. The training script would be very similar to LDM. The major change would be in dataloader where you need to read the segmentation mask as well.