ermongroup/SDEdit

How do i train celeba_hq.ckpt with my own dataset?

Ryhhhh opened this issue · 4 comments

Thanks a lot for your nice work!

How do i train celeba_hq.ckpt with my own dataset?

Hi @Ryhhhh, were you able to figure this out? I also want to train my .ckpt file on my own dataset (CUB). Please help.

Sorry for replying late. I referred to this reply and trained using ddim.
#6 (comment)

Hi @Ryhhhh, were you able to figure this out? I also want to train my .ckpt file on my own dataset (CUB). Please help.

Thanks, @Ryhhhh, for your reply. How long did it take for you to train DDIM and get a checkpoint file?

Thanks, @Ryhhhh, for your reply. How long did it take for you to train DDIM and get a checkpoint file?

I trained on a partial CelebA-HQ dataset for 270,000 steps and achieved good results, but I am not sure whether the model has completely converged.