search

hamadichihaoui / BIRD

This is the official implementation of "Blind Image Restoration via Fast Diffusion Inversion"
211 stars 18 forks source link

How to train my own datasets? #9

Open superY688 opened 2 weeks ago

superY688 commented 2 weeks ago

Hello, I wanted to express my gratitude for your work; it's been instrumental in my current project. However, I've encountered some confusion that I'm hoping you can shed light on. Now，I want to train my own datasets.However,i can not find the related code.Can you provide the code or instruction which i can use to train my datasets. Thank you very much.

hamadichihaoui commented 2 weeks ago

Hi @superY688 and thanks for your interest in our work. Here, we used a pre-trained model. If you want to train your own diffusion model using a custom datasets, you can use OpenAI repo, https://github.com/openai/improved-diffusion.git