PAIR-Diffusion icon indicating copy to clipboard operation
PAIR-Diffusion copied to clipboard

Training procedure

Open CharlesGong12 opened this issue 10 months ago • 1 comments

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?

CharlesGong12 avatar Apr 24 '24 10:04 CharlesGong12

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.

vidit98 avatar May 10 '24 00:05 vidit98