search

ermongroup / SDEdit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
https://sde-image-editing.github.io/
MIT License
966 stars 90 forks source link

How do i train celeba_hq.ckpt with my own dataset？ #17

Open Ryhhhh opened 1 year ago

Ryhhhh commented 1 year ago

Thanks a lot for your nice work!

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

sur-sakthy commented 1 year ago

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

Ryhhhh commented 1 year ago

Sorry for replying late. I referred to this reply and trained using ddim. https://github.com/ermongroup/SDEdit/issues/6#issuecomment-895557419

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

sur-sakthy commented 1 year ago

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

Ryhhhh commented 1 year ago

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.